Update app.py

app.py

@@ -30,7 +30,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
 # Load YOLO model
 # model = YOLO('linhcuem/cham_diem_yolov8')
 # model = YOLO('linhcuem/chamdiemgianhang_yolov8_300epochs')
-# model = YOLO('linhcuem/chamdiemgianhang_yolov8_ver21')
+model = YOLO('linhcuem/chamdiemgianhang_yolov8_ver21')
 # model = YOLO('linhcuem/cham_diem_yolov8_ver20')
 # model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
 
@@ -38,7 +38,7 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
 # current_model_id = model_ids[-1]
 # model = YOLO(current_model_id)
 # model = YOLO(model_path)
-model_path = download_from_hub("linhcuem/checker_TB_yolov8_ver1", "linhcuem/chamdiemgianhang_yolov8_ver21")
+# model_path = download_from_hub("linhcuem/checker_TB_yolov8_ver1", "linhcuem/chamdiemgianhang_yolov8_ver21")
 
 ###################################################
 def yolov8_img_inference(
@@ -50,7 +50,7 @@ def yolov8_img_inference(
     
 ):
     
-    model = YOLO(model_path)
+    # model = YOLO(model_path)
 
     model.conf = conf_threshold
     model.iou = iou_threshold
@@ -134,8 +134,8 @@ interface_image = gr.Interface(
     fn=yolov8_img_inference,
     inputs=[
         gr.Image(type='pil'),
-        gr.Dropdown(["linhcuem/checker_TB_yolov8_ver1", "linhcuem/chamdiemgianhang_yolov8_ver21"], 
-                        default="linhcuem/checker_TB_yolov8_ver1", label="Model"),
+        gr.Dropdown([ "linhcuem/chamdiemgianhang_yolov8_ver21"], 
+                        default="linhcuem/chamdiemgianhang_yolov8_ver21", label="Model"),
         gr.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
         gr.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
         gr.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),