Update README.md

README.md

@@ -25,41 +25,20 @@ Two sources were used for training:
 
 ## Usage
 
-The model can be used as any `ultralytics` model.
-
-The model needs to be downloaded. This is sufficient to perform once.
-
-```
-import os
-import requests
-from ultralytics import YOLO
-
-def download_detector(
-    url="https://huggingface.co/BVRA/TurtleDetector/resolve/main/turtle_detector.pt",
-    path_model="models/turtle_detector.pt",
-    force=False,
-):
-    if not os.path.exists(path_model) or force:
-        os.makedirs(os.path.dirname(path_model), exist_ok=True)
-        r = requests.get(url, timeout=60)
-        r.raise_for_status()
-        with open(path_model, "wb") as f:
-            f.write(r.content)
-
-path_model = "models/turtle_detector.pt"
-download_detector(path_model=path_model)
-```
-
-Once this is done, load the model.
+The model can be used as any `ultralytics` model. First, download and load the model.
 
 ```
+from huggingface_hub import hf_hub_download
 from ultralytics import YOLO
 
-path_model = "models/turtle_detector.pt"
+path_model = hf_hub_download(
+    repo_id="BVRA/TurtleDetector",
+    filename="turtle_detector.pt",
+)
 model = YOLO(path_model)
 ```
 
-Finally, download an image (or use yours) and run the prediction.
+Then download an image (or use yours) and run the prediction.
 
 ```
 import requests