22 lines
863 B
Markdown
22 lines
863 B
Markdown
---
|
|
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")
|
|
``` |