Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -59,15 +59,15 @@ def yolov8_inference(
|
|
59 |
|
60 |
|
61 |
inputs = [
|
62 |
-
gr.
|
63 |
-
gr.
|
64 |
default="yolov8m", label="Model"),
|
65 |
-
gr.
|
66 |
-
gr.
|
67 |
-
gr.
|
68 |
]
|
69 |
|
70 |
-
outputs = gr.
|
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]]
|