Update app.py
Browse files
app.py
CHANGED
@@ -46,12 +46,12 @@ def detect_objects_on_image(image_path, conf_threshold, iou_threshold):
|
|
46 |
demo = gr.Interface(
|
47 |
fn=detect_objects_on_image,
|
48 |
inputs=[
|
49 |
-
gr.
|
50 |
gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence threshold"),
|
51 |
gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU threshold"),
|
52 |
],
|
53 |
outputs=[
|
54 |
-
gr.
|
55 |
],
|
56 |
title="Yolov8 Custom Object Detection",
|
57 |
examples=examples,
|
|
|
46 |
demo = gr.Interface(
|
47 |
fn=detect_objects_on_image,
|
48 |
inputs=[
|
49 |
+
gr.Image(type="filepath", label="Input Image"),
|
50 |
gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence threshold"),
|
51 |
gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU threshold"),
|
52 |
],
|
53 |
outputs=[
|
54 |
+
gr.Image(type="numpy", label="Output Image"),
|
55 |
],
|
56 |
title="Yolov8 Custom Object Detection",
|
57 |
examples=examples,
|