Update README.md

README.md

@@ -85,9 +85,12 @@ This project demonstrates how a compact GPT model can be designed, trained, and
 ```python
 from transformers import GPT2LMHeadModel, GPT2TokenizerFast
 import torch
+import os
 
-model = GPT2LMHeadModel.from_pretrained("Amogh1221/pocketGPT-27M")
-tokenizer = GPT2TokenizerFast.from_pretrained("Amogh1221/pocketGPT-27M")
+os.environ["HUGGINGFACE_HUB_TOKEN"] = "Your Tokens"
+
+model = GPT2LMHeadModel.from_pretrained("Amogh1221/PocketGPT_27M")
+tokenizer = GPT2TokenizerFast.from_pretrained("Amogh1221/PocketGPT_27M")
 
 def ask(prompt):
     formatted = f"<|bos|>Instruction: {prompt}\nResponse:"