Update app.py

app.py

@@ -44,13 +44,13 @@ def attempt_download_from_hub(repo_id, hf_token=None):
     except (RepositoryNotFoundError, HFValidationError):
         return None
 
-MODEL_PATH = attempt_download_from_hub(model_id)
-model = model = YOLO(MODEL_PATH)
-model.to('cuda')
+
 
 @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 = model = YOLO(MODEL_PATH)
+    model.to('cuda')
     results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
     detections = sv.Detections.from_ultralytics(results)