Upload README.md with huggingface_hub

README.md

@@ -20,6 +20,7 @@ A custom GPT model trained on FineWeb 10B dataset.
 
 ## Usage
 
+~~~python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
@@ -34,6 +35,9 @@ tokenizer = AutoTokenizer.from_pretrained("gpt2")
 input_ids = tokenizer("Hello, world", return_tensors="pt").input_ids
 with torch.no_grad():
     outputs = model.generate(input_ids, max_new_tokens=50)
-print(tokenizer.decode(outputs[0]))## Citation
+print(tokenizer.decode(outputs[0]))
+~~~
 
-If you use this model, please cite...
+## Citation
+
+If you use this model, please cite...