linhcuem commited on
Commit
510164d
·
verified ·
1 Parent(s): 1b30cd7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,7 +34,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
34
  # model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
35
  # current_model_id = model_ids[-1]
36
  # model = YOLO(current_model_id)
37
-
38
 
39
  ###################################################
40
  def yolov8_img_inference(
@@ -44,7 +44,7 @@ def yolov8_img_inference(
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
49
  model.iou = iou_threshold
50
  # model.overrides['conf'] = conf_threshold
 
34
  # model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
35
  # current_model_id = model_ids[-1]
36
  # model = YOLO(current_model_id)
37
+ model = YOLO(model_path)
38
 
39
  ###################################################
40
  def yolov8_img_inference(
 
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
49
  model.iou = iou_threshold
50
  # model.overrides['conf'] = conf_threshold