Update app.py

app.py

@@ -29,12 +29,12 @@ 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']
+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')
-# current_model_id = model_ids[-1]
-# model = YOLO(current_model_id)
+current_model_id = model_ids[-1]
+model = YOLO(current_model_id)
 # model = YOLO(model_path)
 
 ###################################################
@@ -46,6 +46,11 @@ def yolov8_img_inference(
     iou_threshold: gr.inputs.Slider = 0.45,
 ):
     # model = YOLO(model_path)
+    global current_model_id
+    global model
+    if model_id != current_model_id:
+        model = YOLO(model_id)
+        current_model_id = model_id
     model.conf = conf_threshold
     model.iou = iou_threshold
     # model.overrides['conf'] = conf_threshold
@@ -137,9 +142,9 @@ interface_image = gr.Interface(
     fn=yolov8_img_inference,
     inputs=[
         gr.inputs.Image(type='pil'),
-        gr.inputs.Dropdown(["linhcuem/chamdiemgianhang_yolov8_ver21"],
-                   default="linhcuem/chamdiemgianhang_yolov8_ver21", label="Model"),
-        
+        # gr.inputs.Dropdown(["linhcuem/chamdiemgianhang_yolov8_ver21"],
+        #            default="linhcuem/chamdiemgianhang_yolov8_ver21", label="Model"),
+        gr.inputs.Dropdown(model_ids, value=model_ids[-1]),
         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"),
         gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),