| # 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 |
| ```python |
| 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) |
| |
