Update README.md

README.md

@@ -9,6 +9,7 @@ Trained on:-
 https://www.kaggle.com/Cornell-University/movie-dialog-corpus
 
 Example:-
+```python
 from transformers import AutoTokenizer, AutoModelWithLMHead
   
 tokenizer = AutoTokenizer.from_pretrained("deepparag/DumBot")
@@ -32,4 +33,5 @@ for step in range(4):
     )
     
     # pretty print last ouput tokens from bot
-    print("DumBot: {}".format(tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)))
+    print("DumBot: {}".format(tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)))
+```