Spaces:

Opsifiz
/

AIB5_Emotion-Classification

Sleeping

Opsifiz commited on Jun 9, 2025

Commit

a362f30

1 Parent(s): bfaffd8

Model V0.2.1

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,10 +15,12 @@ id2label = {
     0: 'Anxiety',
     1: 'BPD',
     2: 'Normal',
-    3: 'bipolar',
-    4: 'depression',
-    5: 'mentalillness',
-    6: 'schizophrenia'
 }
 def clean_text(text):

     0: 'Anxiety',
     1: 'BPD',
     2: 'Normal',
+    3: 'Stress',
+    4: 'Suicidal',
+    5: 'Bipolar',
+    6: 'Depression',
+    7: 'Mentalillness',
+    8: 'Schizophrenia'
 }
 def clean_text(text):