Sa-m commited on
Commit
512f85e
·
1 Parent(s): 3a843b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,7 +6,8 @@ os.system('git clone https://github.com/WongKinYiu/yolov7.git')
6
 
7
  def detect(inp):
8
  os.system('python ./yolov7/detect.py --weights ./yolov7/runs/train/yolov7/weights/best.pt --conf 0.25 --img-size 640 --source f{inp} "--project","./yolov7/runs/detect ')
9
- return f'{inp}'+subprocess.run(["ls","./yolov7/runs/detect/exp"])
 
10
 
11
  inp = gr.inputs.Image(type="filepath", label="Input")
12
  output = gr.outputs.Textbox()
 
6
 
7
  def detect(inp):
8
  os.system('python ./yolov7/detect.py --weights ./yolov7/runs/train/yolov7/weights/best.pt --conf 0.25 --img-size 640 --source f{inp} "--project","./yolov7/runs/detect ')
9
+ otp=inp.split('/')[2]
10
+ return f"./yolov7/runs/detect/exp/{otp}"
11
 
12
  inp = gr.inputs.Image(type="filepath", label="Input")
13
  output = gr.outputs.Textbox()