Update README.md

README.md

@@ -75,13 +75,18 @@ GPT-124M is a lightweight generative language model fine-tuned on the `fineweb-e
 You can use this model for text generation using the `transformers` library.
 
 ```python
+# Import necessary modules from transformers
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 
+# Load tokenizer and model
 model_name = "samkeet/GPT_124M"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
 
+# Create text generation pipeline
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, trust_remote_code=True, device="cpu")
+
+# Generate text
 result = pipe("Earth revolves around the", do_sample=True, max_length=40, temperature=0.9, top_p=0.5, top_k=50)
 print(result)
 ```