linhcuem commited on
Commit
1b30cd7
1 Parent(s): dd5f7b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -38,11 +38,11 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
38
 
39
  ###################################################
40
  def yolov8_img_inference(
41
- image,
42
- model_path= None,
43
- image_size= 640,
44
- conf_threshold= 0.25,
45
- iou_threshold = 0.45,
46
  ):
47
  model = YOLO(model_path)
48
  model.conf = conf_threshold
 
38
 
39
  ###################################################
40
  def yolov8_img_inference(
41
+ image: gr.inputs.Image = None,
42
+ model_path: gr.inputs.Dropdown = None,
43
+ image_size: gr.inputs.Slider = 640,
44
+ conf_threshold: gr.inputs.Slider = 0.25,
45
+ iou_threshold: gr.inputs.Slider = 0.45,
46
  ):
47
  model = YOLO(model_path)
48
  model.conf = conf_threshold