fcakyon commited on
Commit
8f78001
1 Parent(s): 0ca7c89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -31,11 +31,11 @@ model = YOLO(current_model_name)
31
 
32
 
33
  def yolov8_inference(
34
- image: gr.inputs.Image = None,
35
- model_name: gr.inputs.Dropdown = None,
36
- image_size: gr.inputs.Slider = 640,
37
- conf_threshold: gr.inputs.Slider = 0.25,
38
- iou_threshold: gr.inputs.Slider = 0.45,
39
  ):
40
  """
41
  YOLOv8 inference function
 
31
 
32
 
33
  def yolov8_inference(
34
+ image: gr.Image = None,
35
+ model_name: gr.Dropdown = None,
36
+ image_size: gr.Slider = 640,
37
+ conf_threshold: gr.Slider = 0.25,
38
+ iou_threshold: gr.Slider = 0.45,
39
  ):
40
  """
41
  YOLOv8 inference function