Update app.py

app.py

@@ -33,8 +33,8 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
 # 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']
-model_path = download_from_hub('linhcuem/checker_TB_yolov8_ver1')
-model = YOLO(model_path)
+model_path = download_from_hub('linhcuem/checker_TB_yolov8_ver1' , 'linhcuem/chamdiemgianhang_yolov8_ver21')
+# model = YOLO(model_path)
 # current_model_id = model_ids[-1]
 # model = YOLO(current_model_id)
 # model = YOLO(model_path)
@@ -47,7 +47,7 @@ def yolov8_img_inference(
     conf_threshold: gr.inputs.Slider = 0.25,
     iou_threshold: gr.inputs.Slider = 0.45,
 ):
-    # model = YOLO(model_path)
+    model = YOLO(model_path)
 
     model.conf = conf_threshold
     model.iou = iou_threshold
@@ -131,7 +131,7 @@ interface_image = gr.Interface(
     fn=yolov8_img_inference,
     inputs=[
         gr.Image(type='pil'),
-        gr.Dropdown(["linhcuem/checker_TB_yolov8_ver1"], 
+        gr.Dropdown(["linhcuem/checker_TB_yolov8_ver1", "linhcuem/chamdiemgianhang_yolov8_ver21"], 
                         default="linhcuem/checker_TB_yolov8_ver1", 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"),