Update README.md

README.md

@@ -98,7 +98,7 @@ pip install turbot5 -U
 ```
 Then you can import different heads for various purposes; we released more encoder heads for tasks such as token classification, question-answering or text classification and, of course, encoder-decoder heads for conditional generation:
 
-```
+```python
 from turbot5 import T5ForConditionalGeneration
 from turbot5 import T5Config
 from transformers import T5Tokenizer