cxeep commited on
Commit
1f0cc8f
·
verified ·
1 Parent(s): 78c54e7

Update gradio.app.py

Browse files
Files changed (1) hide show
  1. gradio.app.py +2 -2
gradio.app.py CHANGED
@@ -21,8 +21,8 @@ def run_yolov5s_tt100k(input_image_path):
21
  # Create a Gradio interface
22
  iface = gr.Interface(
23
  fn=run_yolov5s_tt100k,
24
- inputs=gr.inputs.Image(type="file"),
25
- outputs=gr.outputs.Image(type="file"),
26
  title="YOLOv5s-TT100K Object Detection",
27
  description="Upload an image to run object detection using YOLOv5s-TT100K."
28
  )
 
21
  # Create a Gradio interface
22
  iface = gr.Interface(
23
  fn=run_yolov5s_tt100k,
24
+ inputs=gr.components.Image(),
25
+ outputs=gr.components.Image(),
26
  title="YOLOv5s-TT100K Object Detection",
27
  description="Upload an image to run object detection using YOLOv5s-TT100K."
28
  )