Update README.md

README.md

@@ -1,4 +1,4 @@
-```
+```python
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
@@ -8,5 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 inputs = tokenizer("Hello my name is", return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=20)
 print(tokenizer.batch_decode(outputs)[0])
-
+"""
+<s> Hello my name is John. I am a 20 year old male. I am a student at a university.
+"""
 ```