Update README.md

README.md

@@ -71,6 +71,7 @@ Users should be aware of the limitations in handling diverse mental health needs
 
 Use the code below to get started with the model.
 
+```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("ayeshaNoor1/Llama_finetunedModel")
@@ -78,6 +79,7 @@ model = AutoModelForCausalLM.from_pretrained("ayeshaNoor1/Llama_finetunedModel")
 
 inputs = tokenizer("Your text here", return_tensors="pt")
 outputs = model.generate(**inputs)
+```
 
 ## Training Details