Spaces:

SudiptaKumar
/

object_detection

Paused

Sudipta Nayak commited on Dec 23, 2023

Commit

5f07b35

1 Parent(s): ff705c3

code changes

Files changed (1) hide show

app/main.py CHANGED Viewed

@@ -47,7 +47,7 @@ async def detect_objects(request: Request, file: UploadFile):
         print('Detect start')
         # Run YOLOv7 detection and save output
-        subprocess.run(["python", "app/detect.py", "--conf", "0.5", "--img-size", "640", "--weights", "app/model/best.pt", "--no-trace",
                         "--source", str(input_file_path), "--save-txt", "--save-conf", "--exist-ok", "--project", str(output_file_path)])
         print('Detect end')

         print('Detect start')
         # Run YOLOv7 detection and save output
+        await subprocess.run(["python", "app/detect.py", "--conf", "0.5", "--img-size", "640", "--weights", "app/model/best.pt", "--no-trace",
                         "--source", str(input_file_path), "--save-txt", "--save-conf", "--exist-ok", "--project", str(output_file_path)])
         print('Detect end')