Spaces:
Sleeping
Sleeping
A newer version of the Streamlit SDK is available:
1.54.0
metadata
title: Emotion Intensity Prediction using Transformer Based Models
emoji: 🤩
colorFrom: purple
colorTo: indigo
sdk: streamlit
sdk_version: 1.46.1
app_file: app.py
pinned: false
Multitask Emotion Prediction Space
This Hugging Face Space hosts a deep learning model that predicts emotions and their intensities from text. It utilizes a BERT-based architecture combined with lexicon features for enhanced performance.
Features:
- BERT-based text understanding.
- Integration of NRC VAD, NRC Emotion Lexicon, and NRC Hashtag Emotion Lexicon.
- Multi-task learning for emotion classification (joy, sadness, anger, fear) and intensity regression.
How to use: Enter your text in the input box below and click "Predict Emotions" to see the model's output.
Model Details:
- Trained on dataset SemEval-2018 El-reg
- Uses
bert-base-uncasedfrom Hugging Face. lex_dim: 21 (number of combined lexicon features)
Files included:
app.py: The Streamlit application code.best_multitask_multilabel_model.pth: Trained model weights.*_scaler.pkl: Joblib-saved feature scalers for lexicon features.NRC-*.txt: Lexicon data files.
Feel free to duplicate this Space and experiment!