Sudipta Nayak
commited on
Commit
·
9d2c349
1
Parent(s):
2e79b76
Code Fix-3
Browse files- app/main.py +1 -1
app/main.py
CHANGED
@@ -35,7 +35,7 @@ async def detect_objects(request: Request, item: Item):
|
|
35 |
file.write(item.file.file.read())
|
36 |
|
37 |
# Run YOLOv7 detection and save output
|
38 |
-
subprocess.run(["python", "detect.py", "--
|
39 |
|
40 |
# Render HTML using Jinja2Templates
|
41 |
return templates.TemplateResponse(
|
|
|
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(
|