aakashv100 commited on
Commit
b7975ba
·
1 Parent(s): d48676a

Refactor app.py to remove unnecessary flag in script command

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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} --save_txt True --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
 
 
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