Update README.md

README.md

@@ -62,6 +62,78 @@ This model is designed for text generation tasks such as:
 
 ---
 
+
+
+#### Load Files from the Repository
+You can use the `from_pretrained` method to load specific files or weights:
+```python
+from transformers import AutoTokenizer, AutoModel
+import torch
+
+# Load the tokenizer (if applicable)
+tokenizer = AutoTokenizer.from_pretrained("RobbiePasquale/gpt-moe-mcts")
+
+# Load model weights
+model = torch.hub.load("RobbiePasquale/gpt-moe-mcts", "GPTWithMoE")
+
+# Use tokenizer and model
+prompt = "Once upon a time in a distant galaxy,"
+input_ids = tokenizer.encode(prompt, return_tensors="pt")
+
+# Generate output (if model is compatible with Hugging Face architecture)
+output = model.generate(input_ids, max_length=50)
+print(tokenizer.decode(output[0]))
+```
+
+---
+
+### **Using `huggingface_hub` for Direct Access to Files**
+The `huggingface_hub` library allows users to download individual files or clone the repository.
+
+#### 1. Install the Library
+```bash
+pip install huggingface_hub
+```
+
+#### 2. Clone the Repository
+You can clone the repository to access all files in the directory:
+```bash
+from huggingface_hub import snapshot_download
+
+# Download repository files
+repo_path = snapshot_download(repo_id="RobbiePasquale/gpt-moe-mcts")
+
+print(f"Repository downloaded to {repo_path}")
+```
+This will download all files from the repository, including:
+- `moe_mcts_new.pt`
+- `q_star.py`
+- `mcts_text_gen.py`
+
+The files will be saved locally in a directory structure matching the repository.
+
+#### 3. Load Specific Files
+To load individual files programmatically:
+```python
+from huggingface_hub import hf_hub_download
+
+# Download specific file
+weights_path = hf_hub_download(repo_id="RobbiePasquale/gpt-moe-mcts", filename="moe_mcts_new.pt")
+print(f"Downloaded weights to {weights_path}")
+```
+
+---
+
+### **Command-Line Access**
+You can also clone the repository using Git:
+```bash
+git lfs install
+git clone https://huggingface.co/RobbiePasquale/gpt-moe-mcts
+```
+
+---
+
+
 ## How to Use
 ### Installation
 Ensure you have the following libraries installed: