cxeep commited on
Commit
2bbfc66
·
verified ·
1 Parent(s): 3806914

Update gradio.app.py

Browse files
Files changed (1) hide show
  1. gradio.app.py +1 -1
gradio.app.py CHANGED
@@ -3,7 +3,7 @@ import subprocess
3
  import os
4
 
5
  def run_yolov5s(image_filepath):
6
- result = subprocess.run(["./yolov5s-tt100k", "-m", "./yolov5s_tt100k_opt_fp32.tmfile", "-i", image_filepath], capture_output=True, text=True)
7
  return "yolov5s-tt100k.out.jpg", result.stdout
8
 
9
  iface = gr.Interface(fn=run_yolov5s, inputs=gr.Image(type="filepath"), outputs=[gr.Image(type="filepath"), gr.Textbox()])
 
3
  import os
4
 
5
  def run_yolov5s(image_filepath):
6
+ result = subprocess.run(["./build/yolov5s-tt100k", "-m", "./yolov5s_tt100k_opt_fp32.tmfile", "-i", image_filepath], capture_output=True, text=True)
7
  return "yolov5s-tt100k.out.jpg", result.stdout
8
 
9
  iface = gr.Interface(fn=run_yolov5s, inputs=gr.Image(type="filepath"), outputs=[gr.Image(type="filepath"), gr.Textbox()])