Update app.py

app.py

@@ -27,12 +27,12 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
              ['test_images/ee106392e56837366e79.jpg','cham_diem_yolov8', 640, 0.25, 0.45], ['test_images/f88d2214a4ee76b02fff.jpg','cham_diem_yolov8', 640, 0.25, 0.45]]
 
 # Load YOLO model
-model = YOLO('linhcuem/cham_diem_yolov8')
+# model = YOLO('linhcuem/cham_diem_yolov8')
 # model = YOLO('linhcuem/chamdiemgianhang_yolov8_300epochs')
 # 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 = YOLO('linhcuem/checker_TB_yolov8_ver1')
+model = YOLO('linhcuem/checker_TB_yolov8_ver1')
 # current_model_id = model_ids[-1]
 # model = YOLO(current_model_id)
 # model = YOLO(model_path)
@@ -137,8 +137,8 @@ interface_image = gr.Interface(
     fn=yolov8_img_inference,
     inputs=[
         gr.inputs.Image(type='pil'),
-        gr.inputs.Dropdown(["linhcuem/cham_diem_yolov8"],
-                   default="linhcuem/cham_diem_yolov8", label="Model"),
+        gr.inputs.Dropdown(["'linhcuem/checker_TB_yolov8_ver1'"],
+                   default="'linhcuem/checker_TB_yolov8_ver1'", label="Model"),
         
         gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
         gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),