Update README.md

README.md

@@ -47,7 +47,7 @@ Refer https://github.com/PraveenSH/RL-Query-Reformulation for more details.
 
 ### How to use
 For optimal utilization of this model, use sampling with repetition penalty to generate diverse samples. Below is the provided sample code.
-```
+```python
 import torch
 from transformers import T5ForConditionalGeneration, T5Tokenizer