Spaces:
Sleeping
Sleeping
Commit ·
b7975ba
1
Parent(s): d48676a
Refactor app.py to remove unnecessary flag in script command
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ import os
|
|
| 7 |
|
| 8 |
def inference(img):
|
| 9 |
name_tag = uuid.uuid4()
|
| 10 |
-
script_command = f"python yolov9/detect_dual.py --source {img} --
|
| 11 |
os.system(script_command)
|
| 12 |
return f"yolov9/runs/detect/{name_tag}/{img.split('/')[-1]}"
|
| 13 |
|
|
|
|
| 7 |
|
| 8 |
def inference(img):
|
| 9 |
name_tag = uuid.uuid4()
|
| 10 |
+
script_command = f"python yolov9/detect_dual.py --source {img} --img 640 --device cpu --weights yolov9/runs/train/exp2/weights/best.pt --name {name_tag}"
|
| 11 |
os.system(script_command)
|
| 12 |
return f"yolov9/runs/detect/{name_tag}/{img.split('/')[-1]}"
|
| 13 |
|