Update app.py

app.py

@@ -31,10 +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_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)
+# 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)
 # model = YOLO(model_path)
 
 ###################################################
@@ -46,18 +46,14 @@ def yolov8_img_inference(
     iou_threshold =  0.45,
 ):
     # model = YOLO(model_path)
-    global current_model_id
-    global model
-    if model_path != current_model_id:
-        model = YOLO(model_path)
-        current_model_id = model_path
+    model = YOLO(model_path)
     model.conf = conf_threshold
     model.iou = iou_threshold
     # model.overrides['conf'] = conf_threshold
     # model.overrides['iou'] = iou_threshold
     # model.overrides['agnostic_nms'] = False
     # model.overrides['max_det'] = 1000
-    # image = read_image
+    image = read_image
     results = model.predict(image, imgsz=image_size, conf=conf_threshold, iou=iou_threshold)
     render = render_result(model=model, image=image, result=results[0])
     # get the model names list
@@ -142,7 +138,8 @@ interface_image = gr.Interface(
     fn=yolov8_img_inference,
     inputs=[
         gr.Image(type='pil'),
-        gr.Dropdown(model_ids, value=model_ids[-1]),
+        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"),
         gr.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),