vumichien commited on
Commit
e8353c3
·
1 Parent(s): 7a064e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -59,15 +59,15 @@ def yolov8_inference(
59
 
60
 
61
  inputs = [
62
- gr.inputs.Image(type="filepath", label="Input Image"),
63
- gr.inputs.Dropdown(["yolov8n", "yolov8m", "yolov8l", "yolov8x"],
64
  default="yolov8m", label="Model"),
65
- gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
66
- gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
67
- gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
68
  ]
69
 
70
- outputs = gr.outputs.Image(type="filepath", label="Output Image")
71
  title = "State-of-the-Art YOLO Models for Object detection"
72
 
73
  examples = [['demo_01.jpg', 'yolov8n', 640, 0.25, 0.45], ['demo_02.jpg', 'yolov8l', 640, 0.25, 0.45], ['demo_03.jpg', 'yolov8x', 1280, 0.25, 0.45]]
 
59
 
60
 
61
  inputs = [
62
+ gr.Image(type="filepath", label="Input Image"),
63
+ gr.Dropdown(["yolov8n", "yolov8m", "yolov8l", "yolov8x"],
64
  default="yolov8m", label="Model"),
65
+ gr.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
66
+ gr.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
67
+ gr.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
68
  ]
69
 
70
+ outputs = gr.Image(type="filepath", label="Output Image")
71
  title = "State-of-the-Art YOLO Models for Object detection"
72
 
73
  examples = [['demo_01.jpg', 'yolov8n', 640, 0.25, 0.45], ['demo_02.jpg', 'yolov8l', 640, 0.25, 0.45], ['demo_03.jpg', 'yolov8x', 1280, 0.25, 0.45]]