emotion-classifier / README.md
Sasivar's picture
Update README.md
6a788f0 verified
|
Raw History Blame Contribute Delete
556 Bytes

Emotion Classifier

This is a fine-tuned DistilBERT model for emotion classification. It can classify emotions into six categories: sadness, joy, love, anger, fear, and surprise.

How to Use

from transformers import AutoModelForSequenceClassification, AutoTokenizer

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

text = "I feel great today!"
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)