AliiaR commited on
Commit
ac5d16a
1 Parent(s): 2a777e6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -8,11 +8,13 @@ tags:
8
  - empathy
9
  - gpt2
10
  ---
11
-
 
12
  ## Training data
13
  It was trained on a large corpus of text, including some emotionally engaging datasets such as the "Facebook Empathetic Dialogues" dataset containing 25k conversations.
14
  A dataset of 25k conversations grounded in emotional situations to facilitate training and evaluating dialogue systems.
15
  You can find a dataset [here](https://www.kaggle.com/datasets/atharvjairath/empathetic-dialogues-facebook-ai).
 
16
  ```python
17
  >>> from transformers import AutoTokenizer, AutoModelForCausalLM
18
  >>> tokenizer = AutoTokenizer.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues")
 
8
  - empathy
9
  - gpt2
10
  ---
11
+ ## Model description
12
+ DialoGPT finetuned on empathetic dialogues
13
  ## Training data
14
  It was trained on a large corpus of text, including some emotionally engaging datasets such as the "Facebook Empathetic Dialogues" dataset containing 25k conversations.
15
  A dataset of 25k conversations grounded in emotional situations to facilitate training and evaluating dialogue systems.
16
  You can find a dataset [here](https://www.kaggle.com/datasets/atharvjairath/empathetic-dialogues-facebook-ai).
17
+ ### How to use
18
  ```python
19
  >>> from transformers import AutoTokenizer, AutoModelForCausalLM
20
  >>> tokenizer = AutoTokenizer.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues")