emotion-classifier

Emotion classification model fine-tuned on emotion dataset.

Model Description

• Model Type: DistilBertForSequenceClassification
• Task: Emotion Classification
• Language: English

Usage

from transformers import AutoModelForSequenceClassification, AutoTokenizer

model = AutoModelForSequenceClassification.from_pretrained("tangxiaoxia/emotion-classifier")
tokenizer = AutoTokenizer.from_pretrained("tangxiaoxia/emotion-classifier")

# Example
text = "I am very happy today!"
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
predictions = outputs.logits.argmax(-1)