Upload README.md with huggingface_hub

README.md

@@ -40,14 +40,22 @@ the paper — see [Performance](#performance) for details.
 ## Quick Start
 
 ```bash
-pip install torch safetensors numpy
+pip install torch safetensors numpy huggingface-hub
 ```
 
 ```python
-from model import AlphaDiffract
-import torch, numpy as np
+import sys
+import torch
+import numpy as np
+from huggingface_hub import snapshot_download
+
+# Download model files
+model_dir = snapshot_download("linked-liszt/OpenAlphaDiffract")
 
-model = AlphaDiffract.from_pretrained(".", device="cpu")
+# Load model
+sys.path.insert(0, model_dir)
+from model import AlphaDiffract
+model = AlphaDiffract.from_pretrained(model_dir, device="cpu")
 
 # 8192-point intensity pattern, normalized to [0, 100]
 pattern = np.load("my_pattern.npy").astype(np.float32)