Model V0.2

app.py

@@ -45,8 +45,8 @@ demo = gr.Interface(
     fn=predict,
     inputs=gr.Textbox(lines=3, placeholder="Enter text here..."),
     outputs=gr.Label(),
-    title="Mental Health Text Classifier",
-    description="Predicts mental health category based on text input."
+    title="Text - Emotion Classifier",
+    description="Predicts emotion category based on text input."
 )
 
 if __name__ == "__main__":

my_model/config.json

@@ -14,7 +14,9 @@
     "3": "LABEL_3",
     "4": "LABEL_4",
     "5": "LABEL_5",
-    "6": "LABEL_6"
+    "6": "LABEL_6",
+    "7": "LABEL_7",
+    "8": "LABEL_8"
   },
   "initializer_range": 0.02,
   "label2id": {
@@ -24,7 +26,9 @@
     "LABEL_3": 3,
     "LABEL_4": 4,
     "LABEL_5": 5,
-    "LABEL_6": 6
+    "LABEL_6": 6,
+    "LABEL_7": 7,
+    "LABEL_8": 8
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",

my_model/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:39a4924aae12999707576d94fab86410a96602f4ad8e7b2c267f6b2eda9ac89a
-size 267847948
+oid sha256:abef957581844df0de6c3969ae39c9a0de7c1a2b33fd938c52342714a5d97c82
+size 267854100