add example for loading the models

README.md

@@ -18,11 +18,26 @@ Please see **paper link**  for details.
 
 ```python
 from pathlib import Path
-from dlclibrary import download_huggingface_model
-# Creates a folder and downloads the model to it
-model_dir = Path("./prima_animal_model_pytorch")
-model_dir.mkdir()
-download_huggingface_model("prima_animal_model_pytorch", model_dir)
+from huggingface_hub import hf_hub_download
+
+repo_id = "MLAdaptiveIntelligence/PRIMA"
+
+model_dir = Path("./prima_model")
+model_dir.mkdir(parents=True, exist_ok=True)
+
+# download stage-1 checkpoint
+s1_path = hf_hub_download(
+    repo_id=repo_id,
+    filename="s1ckpt.ckpt",
+    local_dir=model_dir
+)
+
+# donwload stage-3 checkpoint
+s3_path = hf_hub_download(
+    repo_id=repo_id,
+    filename="s3ckpt.ckpt",
+    local_dir=model_dir
+)
 ```
 ## Intended Use
 • Intended to be used for shape and pose estimation of quadruped images taken from a single view.