Update app.py

app.py

@@ -5,13 +5,8 @@ classifier = pipeline("sentiment-analysis", model="pradanaadn/sucidal-text-class
 
 def text_classification(text):
     label_mapping = {
-    'LABEL_0': 'Anxiety',
-    'LABEL_1': 'Bipolar',
-    'LABEL_2': 'Depression',
-    'LABEL_3': 'Normal',
-    'LABEL_4': 'Personality disorder',
-    'LABEL_5': 'Stress',
-    'LABEL_6': 'Suicidal',
+    'LABEL_0': 'Non-Suicide',
+    'LABEL_1': 'Suicide',
 
 }
     result= classifier(text)