MienOlle commited on
Commit
67007a1
·
verified ·
1 Parent(s): ac6ca58

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +36 -36
README.md CHANGED
@@ -1,37 +1,37 @@
1
- ---
2
- title: Emotion Intensity Prediction using Transformer Based Models
3
- emoji: 🤩
4
- colorFrom: purple
5
- colorTo: indigo
6
- sdk: streamlit
7
- sdk_version: 1.x.x
8
- app_file: app.py
9
- pinned: false
10
- ---
11
-
12
- # Multitask Emotion Prediction Space
13
-
14
- This Hugging Face Space hosts a deep learning model that predicts emotions and their intensities from text.
15
- It utilizes a BERT-based architecture combined with lexicon features for enhanced performance.
16
-
17
- **Features:**
18
- - BERT-based text understanding.
19
- - Integration of NRC VAD, NRC Emotion Lexicon, and NRC Hashtag Emotion Lexicon.
20
- - Multi-task learning for emotion classification (joy, sadness, anger, fear) and intensity regression.
21
-
22
- **How to use:**
23
- Enter your text in the input box below and click "Predict Emotions" to see the model's output.
24
-
25
- **Model Details:**
26
- - Trained on dataset SemEval-2018 El-reg
27
- - Uses `bert-base-uncased` from Hugging Face.
28
- - `lex_dim`: 21 (number of combined lexicon features)
29
-
30
- **Files included:**
31
- - `app.py`: The Streamlit application code.
32
- - `best_multitask_multilabel_model.pth`: Trained model weights.
33
- - `*_scaler.pkl`: Joblib-saved feature scalers for lexicon features.
34
- - `NRC-*.txt`: Lexicon data files.
35
-
36
- ---
37
  Feel free to duplicate this Space and experiment!
 
1
+ ---
2
+ title: Emotion Intensity Prediction using Transformer Based Models
3
+ emoji: 🤩
4
+ colorFrom: purple
5
+ colorTo: indigo
6
+ sdk: streamlit
7
+ sdk_version: 1.46.1
8
+ app_file: app.py
9
+ pinned: false
10
+ ---
11
+
12
+ # Multitask Emotion Prediction Space
13
+
14
+ This Hugging Face Space hosts a deep learning model that predicts emotions and their intensities from text.
15
+ It utilizes a BERT-based architecture combined with lexicon features for enhanced performance.
16
+
17
+ **Features:**
18
+ - BERT-based text understanding.
19
+ - Integration of NRC VAD, NRC Emotion Lexicon, and NRC Hashtag Emotion Lexicon.
20
+ - Multi-task learning for emotion classification (joy, sadness, anger, fear) and intensity regression.
21
+
22
+ **How to use:**
23
+ Enter your text in the input box below and click "Predict Emotions" to see the model's output.
24
+
25
+ **Model Details:**
26
+ - Trained on dataset SemEval-2018 El-reg
27
+ - Uses `bert-base-uncased` from Hugging Face.
28
+ - `lex_dim`: 21 (number of combined lexicon features)
29
+
30
+ **Files included:**
31
+ - `app.py`: The Streamlit application code.
32
+ - `best_multitask_multilabel_model.pth`: Trained model weights.
33
+ - `*_scaler.pkl`: Joblib-saved feature scalers for lexicon features.
34
+ - `NRC-*.txt`: Lexicon data files.
35
+
36
+ ---
37
  Feel free to duplicate this Space and experiment!