| from transformers import pipeline | |
| classifier = pipeline("text-classification", | |
| model='bhadresh-savani/distilbert-base-uncased-emotion', | |
| return_all_scores=True) | |
| def get_emotion(text='No text yet'): | |
| prediction = classifier(text)[0] | |
| result = max(prediction, key=lambda x: x['score']) | |
| return result | |