linhcuem commited on
Commit
4b41529
·
verified ·
1 Parent(s): e97ca08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,7 +33,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
33
  # model = YOLO('linhcuem/chamdiemgianhang_yolov8_ver21')
34
  # model = YOLO('linhcuem/cham_diem_yolov8_ver20')
35
  # model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
36
- model_path = download_from_hub('linhcuem/checker_TB_yolov8_ver1' , 'linhcuem/chamdiemgianhang_yolov8_ver21')
37
  # model = YOLO(model_path)
38
  # current_model_id = model_ids[-1]
39
  # model = YOLO(current_model_id)
@@ -47,6 +47,7 @@ def yolov8_img_inference(
47
  conf_threshold: gr.inputs.Slider = 0.25,
48
  iou_threshold: gr.inputs.Slider = 0.45,
49
  ):
 
50
  model = YOLO(model_path)
51
 
52
  model.conf = conf_threshold
 
33
  # model = YOLO('linhcuem/chamdiemgianhang_yolov8_ver21')
34
  # model = YOLO('linhcuem/cham_diem_yolov8_ver20')
35
  # model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
36
+
37
  # model = YOLO(model_path)
38
  # current_model_id = model_ids[-1]
39
  # model = YOLO(current_model_id)
 
47
  conf_threshold: gr.inputs.Slider = 0.25,
48
  iou_threshold: gr.inputs.Slider = 0.45,
49
  ):
50
+ model_path = download_from_hub(model_path)
51
  model = YOLO(model_path)
52
 
53
  model.conf = conf_threshold