Sudipta Nayak
commited on
Commit
·
3ebd9c2
1
Parent(s):
695d423
print stsmt added
Browse files- app/main.py +6 -0
app/main.py
CHANGED
@@ -27,6 +27,8 @@ async def root(request: Request):
|
|
27 |
|
28 |
@app.post("/detect", response_class=HTMLResponse)
|
29 |
async def detect_objects(request: Request, item: Item):
|
|
|
|
|
30 |
input_file = f"uploads/{item.file.filename}"
|
31 |
output_file = f"static/output/{item.file.filename}"
|
32 |
|
@@ -34,9 +36,13 @@ async def detect_objects(request: Request, item: Item):
|
|
34 |
with open(input_file, "wb") as file:
|
35 |
file.write(item.file.file.read())
|
36 |
|
|
|
|
|
37 |
# Run YOLOv7 detection and save output
|
38 |
subprocess.run(["python", "detect.py", "--conf", "0.5", "--img-size", "640", "--weights", "app/model/best.pt", "--source", input_file, "--save-txt", "--save-conf", "--exist-ok", "--project", output_file])
|
39 |
|
|
|
|
|
40 |
# Render HTML using Jinja2Templates
|
41 |
return templates.TemplateResponse(
|
42 |
"result.html",
|
|
|
27 |
|
28 |
@app.post("/detect", response_class=HTMLResponse)
|
29 |
async def detect_objects(request: Request, item: Item):
|
30 |
+
print('File name:', item.file.filename)
|
31 |
+
|
32 |
input_file = f"uploads/{item.file.filename}"
|
33 |
output_file = f"static/output/{item.file.filename}"
|
34 |
|
|
|
36 |
with open(input_file, "wb") as file:
|
37 |
file.write(item.file.file.read())
|
38 |
|
39 |
+
print('Detect start')
|
40 |
+
|
41 |
# Run YOLOv7 detection and save output
|
42 |
subprocess.run(["python", "detect.py", "--conf", "0.5", "--img-size", "640", "--weights", "app/model/best.pt", "--source", input_file, "--save-txt", "--save-conf", "--exist-ok", "--project", output_file])
|
43 |
|
44 |
+
print('Detect end')
|
45 |
+
|
46 |
# Render HTML using Jinja2Templates
|
47 |
return templates.TemplateResponse(
|
48 |
"result.html",
|