Update README.md

README.md

@@ -12,9 +12,9 @@ tags:
 
 It was trained on a large corpus of text, including some emotionally engaging datasets such as the "Facebook Empathetic Dialogues" dataset containing 25k conversations.
 A dataset of 25k conversations grounded in emotional situations to facilitate training and evaluating dialogue systems.
+```python
+>>> from transformers import AutoTokenizer, AutoModelForCausalLM
 
-from transformers import AutoTokenizer, AutoModelForCausalLM
+>>> tokenizer = AutoTokenizer.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues")
 
-tokenizer = AutoTokenizer.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues")
-
-model = AutoModelForCausalLM.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues")
+>>> model = AutoModelForCausalLM.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues")