MarwanSalim023/emotion-detector

AI Emotion Detector

DistilBERT fine-tuned (transfer learning) on the dair-ai/emotion dataset to classify English text into 6 emotions: sadness, joy, love, anger, fear, surprise.

• Base model: distilbert-base-uncased
• Dataset: dair-ai/emotion (16,000 train / 2,000 validation / 2,000 test)
• Training: 2 epochs, batch size 32, learning rate 5e-5, Google Colab T4 GPU
• Test accuracy: 93.00%
• Live app: https://ai-emotion-detector-mehrwan.streamlit.app

Usage

from transformers import pipeline
clf = pipeline("text-classification", model="MarwanSalim023/emotion-detector")
print(clf("I just got my first job offer!"))