Update README.md

README.md

@@ -47,7 +47,7 @@ For the full implementation and instructions, please see our official GitHub rep
 
 To use the pretrained models from this hub, you can use the `huggingface_hub` library to download the files into the correct directory structure within your local `pretrained_models` folder.
 
-**Example: Download All Models (Terminal Command)**
+**Example: Using the Terminal (Downloads all models at once)**
 
 Make sure you have `huggingface-hub` installed (`pip install huggingface-hub`). Then run the following command from your terminal:
 
@@ -57,6 +57,20 @@ huggingface-cli download Moon-bow/DPoser-X --repo-type model --local-dir pretrai
 
 This command will download the entire repository contents into a local folder named `pretrained_models`, preserving the required directory structure. You can then proceed with the instructions in our GitHub repository.
 
+**Example: Within Python Code (Automatic Download)**
+
+You can also use the `huggingface_hub` library to download the models programmatically:
+
+```python
+from huggingface_hub import snapshot_download, hf_hub_download
+
+# download entire repo
+filepath = snapshot_download(repo_id="Moon-bow/DPoser-X")
+
+# download one file
+filepath = hf_hub_download(repo_id="Moon-bow/DPoser-X", filename="body/BaseMLP/last.ckpt")
+```
+
 ## Citation
 If you find our work useful, please cite our paper: