Fix Quick Start: use snapshot_download for proper import

README.md

@@ -40,10 +40,17 @@ pip install scikit-learn numpy joblib huggingface_hub
 ### Load Pre-trained Checkpoints
 
 ```python
+from huggingface_hub import snapshot_download
+import sys
+
+# Download model
+path = snapshot_download("JiaqiXue/r2-router")
+sys.path.insert(0, path)
+
 from router import R2Router
 
 # Load pre-trained KNN checkpoints (no training needed)
-router = R2Router.from_pretrained("jqxue1999/r2-router")
+router = R2Router.from_pretrained(path)
 
 # Route a query (requires 1024-dim embedding from Qwen3-0.6B)
 result = router.route(embedding)
@@ -55,10 +62,16 @@ print(f"Predicted Quality: {result['predicted_quality']:.3f}")
 ### Train from Scratch
 
 ```python
+from huggingface_hub import snapshot_download
+import sys
+
+path = snapshot_download("JiaqiXue/r2-router")
+sys.path.insert(0, path)
+
 from router import R2Router
 
 # Train KNN from the provided sub_10 training data
-router = R2Router.from_training_data("jqxue1999/r2-router", k=80)
+router = R2Router.from_training_data(path, k=80)
 
 # Route a query
 result = router.route(embedding)