cxeep commited on
Commit
607d55d
·
verified ·
1 Parent(s): f96091f

Update gradio.app.py

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