File size: 863 Bytes
d986968 1b38f52 ac5d16a 2a777e6 1b38f52 af7c201 2a777e6 ac5d16a e7cd5b0 12f7d19 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
---
language:
- en
pipeline_tag: conversational
tags:
- psychology
- dialogues
- 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")
>>> model = AutoModelForCausalLM.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues")
``` |