Spaces:

Detomo
/

Object_detection

Running

vumichien commited on Mar 15, 2023

Commit

5a887dc

1 Parent(s): e8353c3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,11 +6,11 @@ from ultralyticsplus import YOLO
 def yolov8_inference(
-    image: gr.inputs.Image = None,
-    model_path: gr.inputs.Dropdown = None,
-    image_size: gr.inputs.Slider = 640,
-    conf_threshold: gr.inputs.Slider = 0.25,
-    iou_threshold: gr.inputs.Slider = 0.45,
 ):
     """
     YOLOv8 inference function

 def yolov8_inference(
+    image,
+    model_path,
+    image_size,
+    conf_threshold,
+    iou_threshold,
 ):
     """
     YOLOv8 inference function