Update app.py

app.py

@@ -49,6 +49,7 @@ def attempt_download_from_hub(repo_id, hf_token=None):
 @spaces.GPU(duration=10)
 def LeYOLO_inference(image, model_id, image_size, conf_threshold, iou_threshold):
     # MODEL_PATH = attempt_download_from_hub(model_id)
+    model = YOLO.from_pretrained(model_id)
     split_path = model.split('/')
     if len(split_path) == 2 and (not os.path.exists(model)):
         model = YOLO.from_pretrained(model_id)