linhcuem commited on
Commit
e8d4f9d
1 Parent(s): 151e108

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -33,11 +33,11 @@ model = YOLO('linhcuem/cham_diem_yolov8')
33
  def yolov8_img_inference(
34
  image: gr.inputs.Image = None,
35
  model_path: 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
- model = YOLO(model_path)
41
  model.conf = conf_threshold
42
  model.iou = iou_threshold
43
  # model.overrides['conf'] = conf_threshold
 
33
  def yolov8_img_inference(
34
  image: gr.inputs.Image = None,
35
  model_path: gr.inputs.Dropdown = None,
36
+ image_size = 640,
37
+ conf_threshold = 0.25,
38
+ iou_threshold = 0.45,
39
  ):
40
+ # model = YOLO(model_path)
41
  model.conf = conf_threshold
42
  model.iou = iou_threshold
43
  # model.overrides['conf'] = conf_threshold