Views
No views yet
1
2from transformers import pipeline
3model_id = "suvrobaner/distilbert-base-uncased-finetuned-emotion-en-tweets"
4classifier = pipeline("text-classification", model = model_id)
5
6custom_tweet = "I saw a movie today and it was really good."
7preds = classifier(custom_tweet, return_all_scores=True)
8
9labels = ['sadness', 'joy', 'love', 'anger', 'fear', 'surprise']
10
11preds_df = pd.DataFrame(preds[0])
12
13import matplotlib.pyplot as plt
14plt.bar(labels, 100 * preds_df["score"], color='C0')
15plt.title(f'"{custom_tweet}"')
16plt.ylabel("Class probability (%)")
17plt.show()