Update README.md

README.md

@@ -36,7 +36,8 @@ Additional resources:
 
 
 Using the model in a text-generation pipeline:
-```
+
+```python
 from transformers import pipeline
 import torch
 
@@ -55,7 +56,8 @@ print(outputs[0]["generated_text"][len(prompt):])
 
 
 Using the model in a conversational pipeline (works with transformers==4.36.2, issues with output in newer versions):
-```
+
+```python
 from transformers import pipeline, Conversation
 import torch