linhcuem commited on
Commit
55bb760
1 Parent(s): 9acf0ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -38,6 +38,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
38
  # current_model_id = model_ids[-1]
39
  # model = YOLO(current_model_id)
40
  # model = YOLO(model_path)
 
41
 
42
  ###################################################
43
  def yolov8_img_inference(
@@ -46,9 +47,9 @@ def yolov8_img_inference(
46
  image_size: gr.inputs.Slider = 640,
47
  conf_threshold: gr.inputs.Slider = 0.25,
48
  iou_threshold: gr.inputs.Slider = 0.45,
49
- token: gr.inputs.Textbox = None
50
  ):
51
- model_path = download_from_hub(model_path, token)
52
  model = YOLO(model_path)
53
 
54
  model.conf = conf_threshold
 
38
  # current_model_id = model_ids[-1]
39
  # model = YOLO(current_model_id)
40
  # model = YOLO(model_path)
41
+ model_path = download_from_hub("linhcuem/checker_TB_yolov8_ver1", "linhcuem/chamdiemgianhang_yolov8_ver21")
42
 
43
  ###################################################
44
  def yolov8_img_inference(
 
47
  image_size: gr.inputs.Slider = 640,
48
  conf_threshold: gr.inputs.Slider = 0.25,
49
  iou_threshold: gr.inputs.Slider = 0.45,
50
+
51
  ):
52
+
53
  model = YOLO(model_path)
54
 
55
  model.conf = conf_threshold