Upload README.md

README.md

@@ -67,7 +67,48 @@ generated_text = generate_text(
 print(generated_text)
 ```
 
-### Method 2: Module-specific imports
+### Method 2: Using snapshot_download
+```python
+from huggingface_hub import snapshot_download
+import sys
+import torch, json
+
+# Download the entire repository
+repo_id = "sky-2002/deepseek-tinystories-60M"
+repo_dir = snapshot_download(repo_id)
+
+# Import the local package from the downloaded repo
+sys.path.append(str(repo_dir))
+
+from deepseek_tinystories import DeepseekInspiredModel, DeepSeekModelConfig, TinyStoriesProcesssor, generate_text
+
+# Load config & model
+config_path = f"{repo_dir}/config.json"
+model_path = f"{repo_dir}/pytorch_model.bin"
+
+config = DeepSeekModelConfig(**json.load(open(config_path)))
+model = DeepseekInspiredModel(config)
+model.load_state_dict(torch.load(model_path, map_location="cpu"))
+model.eval()
+
+# Initialize processor
+processor = TinyStoriesProcesssor()
+
+# Generate text
+prompt = "Once upon a time, there was a little girl..."
+generated_text = generate_text(
+    model=model,
+    data_processor=processor,
+    prompt=prompt,
+    max_new_tokens=50,
+    temperature=0.8,
+    top_k=40,
+    device="cpu"
+)
+print(generated_text)
+```
+
+### Method 3: Module-specific imports
 ```python
 from deepseek_tinystories.modeling_deepseek import DeepseekInspiredModel, DeepSeekModelConfig
 from deepseek_tinystories.processor import TinyStoriesProcesssor