Update README.md

README.md

@@ -31,11 +31,14 @@ Final test Loss: 2.7963
 
 _________________________________________
 
-try the following script for inference:
+## Inference
 
-!pip install huggingface_hub
-!pip install transformers
-!pip install torch
+
+"pip install huggingface_hub"
+"pip install transformers"
+"pip install torch"
+
+```python
 from transformers import GPT2Tokenizer, GPT2Config, GPT2LMHeadModel
 from huggingface_hub import hf_hub_download
 import torch
@@ -91,3 +94,4 @@ while True:
         break
     output = generate_text(prompt)
     print(f"Generated text: {output}")
+```