Update app.py

app.py

@@ -1,16 +1,26 @@
 import gradio as gr
 from transformers import pipeline
 
-classifier = pipeline("sentiment-analysis", model="Tirendaz/my_distilbert_model")
+classifier = pipeline("sentiment-analysis", model="pradanaadn/sucidal-text-classification-distillbert")
 
 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',
+
+}
     result= classifier(text)
-    sentiment_label = result[0]['label']
+    sentiment_label = label_mapping[result[0]['label']]
     sentiment_score = result[0]['score']
     formatted_output = f"This sentiment is {sentiment_label} with the probability {sentiment_score*100:.2f}%"
     return formatted_output
 
-examples=["This is wonderful movie!", "The movie was really bad; I didn't like it."]
+examples=["It's been a rollercoaster lately. One moment I'm on top of the world, full of energy, and the next, I'm down in the dumps, feeling hopeless", "I've noticed that my reactions and feelings often don't match what others expect. It creates tension in my relationships, and sometimes I feel like I'm not in control of myself."]
 
 io = gr.Interface(fn=text_classification, 
                          inputs= gr.Textbox(lines=2, label="Text", placeholder="Enter title here..."), 
@@ -18,5 +28,5 @@ io = gr.Interface(fn=text_classification,
                          title="Text Classification",
                          description="Enter a text and see the text classification result!",
                          examples=examples)
-print('test')
+
 io.launch()