Update app.py

app.py

@@ -31,7 +31,10 @@ image_path = [['test_images/2a998cfb0901db5f8210.jpg','cham_diem_yolov8', 640, 0
 # model = YOLO('linhcuem/chamdiemgianhang_yolov8_300epochs')
 # model = YOLO('linhcuem/chamdiemgianhang_yolov8_ver21')
 # model = YOLO('linhcuem/cham_diem_yolov8_ver20')
-model = YOLO('linhcuem/checker_TB_yolov8_ver1')
+model_ids = ['linhcuem/checker_TB_yolov8_ver1', 'linhcuem/cham_diem_yolov8', 'linhcuem/chamdiemgianhang_yolov8_300epochs', 'linhcuem/cham_diem_yolov8_ver20', 'linhcuem/chamdiemgianhang_yolov8_ver21']
+current_model_id = model_ids[-1]
+model = YOLO(current_model_id)
+
 
 ###################################################
 def yolov8_img_inference(
@@ -149,8 +152,9 @@ interface_vid = gr.Interface(
 inputs_image = [
     # gr.inputs.Image(type="filepath", label="Input Image"),
     gr.Image(type="pil"),
-    gr.Dropdown(["linhcuem/linhcuem/chamdiemgianhang_yolov8_ver1"], 
+    # gr.Dropdown(["linhcuem/linhcuem/chamdiemgianhang_yolov8_ver1"], 
                         default="linhcuem/chamdiemgianhang_yolov8_ver1", label="Model"),
+    
     gr.Slider(maximum=1280, step=32, value = 640, label="Image Size"),
     gr.Slider(maximum=1.0 , step=0.05, value = 0.25, label="Confidence Threshold"),
     gr.Slider(maximum=1.0, step=0.05, value = 0.45, label="IOU Threshold"),
@@ -167,8 +171,9 @@ interface_image = gr.Interface(
     fn=yolov8_img_inference,
     inputs=[
         gr.Image(type='pil'),
-        gr.Dropdown(["linhcuem/chamdiemgianhang_yolov8_ver1"],
-                   default="linhcuem/chamdiemgianhang_yolov8_ver1"),
+        # gr.Dropdown(["linhcuem/chamdiemgianhang_yolov8_ver1"],
+        #            default="linhcuem/chamdiemgianhang_yolov8_ver1"),
+        gr.Dropdown(model_ids, value=model_ids[-1]),
         gr.Slider(maximum=1280, step=32, value=640),
         gr.Slider(maximum=1.0, step=0.05, value=0.25),
         gr.Slider(maximum=1.0, step=0.05, value=0.45),