Update README.md

README.md

@@ -26,7 +26,26 @@ from ultralytics import YOLOv10
 import supervision as sv
 import cv2
 
-MODEL_PATH = 'yolov10n.pt'
+def attempt_download_from_hub(repo_id, hf_token=None):
+    # https://github.com/fcakyon/yolov5-pip/blob/main/yolov5/utils/downloads.py
+    from huggingface_hub import hf_hub_download, list_repo_files
+    from huggingface_hub.utils._errors import RepositoryNotFoundError
+    from huggingface_hub.utils._validators import HFValidationError
+    try:
+        repo_files = list_repo_files(repo_id=repo_id, repo_type='model', token=hf_token)
+        model_file = [f for f in repo_files if f.endswith('.pt')][0]
+        file = hf_hub_download(
+            repo_id=repo_id,
+            filename=model_file,
+            repo_type='model',
+            token=hf_token,
+        )
+        return file
+    except (RepositoryNotFoundError, HFValidationError):
+        return None
+    
+
+MODEL_PATH = attempt_download_from_hub("kadirnar/yolov10n", hf_token="hf_token")
 IMAGE_PATH = 'dog.jpeg'
 
 model = YOLOv10(MODEL_PATH)