Update README.md

README.md

@@ -41,12 +41,16 @@ Checkpoints for **“Text-to-CT Generation via 3D Latent Diffusion Model with Co
 
 ### How to Get Started (Python)
 ```python
-from huggingface_hub import hf_hub_download
-repo_id = "yourname/text2ct-weights"  # replace with the actual repo id
+from huggingface_hub import snapshot_download
 
-autoencoder_path = hf_hub_download(repo_id, "autoencoder_epoch273.pt")
-unet_path = hf_hub_download(repo_id, "unet_rflow_200ep.pt")
-clip_path = hf_hub_download(repo_id, "CLIP3D_Finding_Impression_30ep.pt")
+repo_id = "dmolino/text2ct-weights"
+
+snapshot_download(
+    repo_id=repo_id,
+    repo_type="model",
+    local_dir="your_local_path" 
+)
+```
 
 # Use these in the code release configs:
 # trained_autoencoder_path -> autoencoder_path