Update app.py

app.py

@@ -11,18 +11,18 @@ from ultralyticsplus import YOLO
 
 app_title = "Detect defects in bird nest jar"
 models_ids = ['linhcuem/defects_bird_nest_jar_yolov5m6']
-DEFAULT_DET_MODEL_ID_yolov5 = 'linhcuem/defects_bird_nest_jar_yolov5m6'
-DEFAULT_DET_MODEL_ID_yolov8 = 'linhcuem/defects_bird_nest_jar_yolov8'
+# DEFAULT_DET_MODEL_ID_yolov5 = 'linhcuem/defects_bird_nest_jar_yolov5m6'
+# DEFAULT_DET_MODEL_ID_yolov8 = 'linhcuem/defects_bird_nest_jar_yolov8'
 
 current_model_id = models_ids[-1]
 model = yolov5.load(current_model_id)
-model_yolov8 = YOLO(DEFAULT_DET_MODEL_ID_yolov8)
+# model_yolov8 = YOLO(DEFAULT_DET_MODEL_ID_yolov8)
 examples = [['test_images/16823291638707408-a2A2448-23gmBAS_40174045.jpg', 0.25, 'linhcuem/defects_nest_jar_yolov5'], ['test_images/16823292102253310-a2A2448-23gmBAS_40174046.jpg', 0.25, 'linhcuem/defects_nest_jar_yolov5'], ['test_images/16823291808953550-a2A2448-23gmBAS_40174048.jpg', 0.25, 'linhcuem/defects_nest_jar_yolov5'], ['test_images/16823291801532480-a2A2448-23gmBAS_40174048.jpg', 0.25, 'linhcuem/defects_nest_jar_yolov5']]
 
 def predict(image, threshold=0.25, model_id=None):
     #update model if required
     global current_model_id
-    global model_yolov5, model_yolov8
+    global model
     if model_id != current_model_id:
         model = yolov5.load(models_id)
         # model_yolov8 = YOLO(DEFAULT_DET_MODEL_ID_yolov8)