Update README.md

README.md

@@ -40,13 +40,16 @@ library_name: pytorch
 
 ### Download Checkpoint
 
-```python
+```
+python -c "
 from huggingface_hub import hf_hub_download
-
-checkpoint_path = hf_hub_download(
-    repo_id="Hila/selfflow-imagenet256",
-    filename="selfflow_imagenet256.pt"
+hf_hub_download(
+    repo_id='Hila/Self-Flow',
+    filename='selfflow_imagenet256.pt',
+    local_dir='./checkpoints'
 )
+print('Downloaded!')
+"
 ```
 and follow the instructions in our repository: https://github.com/black-forest-labs/Self-Flow