app.py

import gradio as gr
from transformers import pipeline

classifier = pipeline("sentiment-analysis", model="pradanaadn/sucidal-text-classification-distillbert")

def text_classification(text):
    label_mapping = {
    'LABEL_0': 'Non-Suicide',
    'LABEL_1': 'Suicide',

}
    result= classifier(text)
    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=["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 can't keep going like this anymore. Everything feels hopeless, and I don't see a way out. I feel like the world would be better off without me."]

io = gr.Interface(fn=text_classification, 
                         inputs= gr.Textbox(lines=2, label="Text", placeholder="Enter title here..."), 
                         outputs=gr.Textbox(lines=2, label="Text Classification Result"),
                         title="Suicidal Intent Detection",
                         description="Enter a text and see the text classification result!",
                         examples=examples)

io.launch()