File size: 354 Bytes
1a62c90 d801d73 1a62c90 d801d73 9f6e2f8 1a62c90 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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
|