Update README.md

add "python" into code lines to show it in a better way.

README.md

@@ -25,7 +25,7 @@ The dataset used is [OpenAssistant/oasst_top1_2023-08-25](https://huggingface.co
 #### How to use
 You will need the transformers>=4.31
 Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for more information.
-```
+```python
 from transformers import AutoTokenizer
 import transformers 
 import torch