|
--- |
|
title: Emotion Detection App |
|
emoji: π |
|
colorFrom: indigo |
|
colorTo: pink |
|
sdk: gradio |
|
sdk_version: "4.27.0" |
|
app_file: app.py |
|
pinned: false |
|
--- |
|
|
|
# π Emotion Detection from Text using BERT |
|
|
|
Welcome to the **Emotion Detection Web App**! This application uses a fine-tuned BERT model to detect human emotions from short pieces of text. |
|
|
|
--- |
|
|
|
## π Demo |
|
|
|
π Try the live app: [Click here to open the web app](https://huggingface.co/spaces/sujith13082003/emotion_detection) |
|
|
|
--- |
|
|
|
## π Description |
|
|
|
This project leverages the `nateraw/bert-base-uncased-emotion` model from Hugging Face Transformers to classify input text into one of six emotions: |
|
|
|
- π’ Sadness |
|
- π Joy |
|
- π Love |
|
- π‘ Anger |
|
- π± Fear |
|
- π² Surprise |
|
|
|
It uses: |
|
- **Hugging Face Transformers** for model and tokenizer |
|
- **PyTorch** for deep learning inference |
|
- **Gradio** to build an interactive web interface |
|
|
|
--- |
|
|
|
## π§ Model Used |
|
|
|
- **Model Name**: `nateraw/bert-base-uncased-emotion` |
|
- **Base Architecture**: BERT (uncased) |
|
- **Dataset**: GoEmotions subset |
|
|
|
--- |
|
|
|
## π¦ Dependencies |
|
|
|
Dependencies are defined in `requirements.txt`: |
|
- `transformers` |
|
- `torch` |
|
- `gradio` |
|
|
|
--- |
|
|
|
## π Use Cases |
|
|
|
- Social media sentiment analysis |
|
- Customer feedback classification |
|
- Chatbot emotion understanding |
|
- Mental health applications |
|
|
|
--- |
|
|
|
## π¨βπ» Author |
|
|
|
- **Sujith Kumar** |
|
- Hugging Face: [@sujith13082003](https://huggingface.co/sujith13082003) |
|
|
|
--- |
|
|
|
## π License |
|
|
|
This project is for educational and research purposes. Refer to individual library licenses for commercial use. |
|
|