Update README.md

README.md

@@ -8,11 +8,13 @@ tags:
 - empathy
 - gpt2
 ---
-
+## Model description
+DialoGPT finetuned on empathetic dialogues
 ## Training data
 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.
 You can find a dataset [here](https://www.kaggle.com/datasets/atharvjairath/empathetic-dialogues-facebook-ai).
+### How to use
 ```python
 >>> from transformers import AutoTokenizer, AutoModelForCausalLM
 >>> tokenizer = AutoTokenizer.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues")