Sudipta Nayak commited on
Commit
9d2c349
·
1 Parent(s): 2e79b76

Code Fix-3

Browse files
Files changed (1) hide show
  1. 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", "--weights", "model/best.pt", "--img-size", "640", "--source", input_file, "--save-txt", "--save-conf", "--exist-ok", "--output", output_file])
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(