Update app.py

app.py

@@ -48,8 +48,8 @@ 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 = model = YOLO(MODEL_PATH)
+    # MODEL_PATH = attempt_download_from_hub(model_id)
+    model = model = YOLO.from_pretrained(model_id)
     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)
@@ -77,12 +77,12 @@ def app():
                 model_id = gr.Dropdown(
                     label="Model",
                     choices=[
-                        "kadirnar/LeYOLOSmall",
-                        "kadirnar/LeYOLONano",
-                        "kadirnar/LeYOLOMedium",
-                        "kadirnar/LeYOLOLarge",
+                        "lhollard/leyolo-nano",
+                        "lhollard/leyolo-small",
+                        "lhollard/leyolo-medium",
+                        "lhollard/leyolo-large",
                     ],
-                    value="kadirnar/LeYOLOMedium",
+                    value="lhollard/leyolo-medium",
                 )
                 image_size = gr.Slider(
                     label="Image Size",
@@ -126,14 +126,14 @@ def app():
             examples=[
                 [
                     "dog.jpeg",
-                    "kadirnar/LeYOLOMedium",
+                    "lhollard/leyolo-medium",
                     640,
                     0.25,
                     0.45,
                 ],
                 [
                     "zidane.jpg",
-                    "kadirnar/LeYOLOMedium",
+                    "lhollard/leyolo-medium",
                     640,
                     0.25,
                     0.45,