pytorch distilbert text-classification en dataset:emotion transformers license:apache-2.0
---
language: en
tags:
- text-classification
- pytorch
license: apache-2.0
datasets:
- emotion
---


```python

from transformers import pipeline
model_id = "suvrobaner/distilbert-base-uncased-finetuned-emotion-en-tweets"
classifier = pipeline("text-classification", model = model_id)

custom_tweet = "I saw a movie today and it was really good."
preds = classifier(custom_tweet, return_all_scores=True)

labels = ['sadness', 'joy', 'love', 'anger', 'fear', 'surprise']

preds_df = pd.DataFrame(preds[0])

import matplotlib.pyplot as plt
plt.bar(labels, 100 * preds_df["score"], color='C0')
plt.title(f'"{custom_tweet}"')
plt.ylabel("Class probability (%)")
plt.show()
```
