Update app.py

app.py

@@ -1,3 +1,26 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.load("models/wcyat/bert-suicide-detection-hk").launch()
+# Load pre-trained sentiment-analysis pipeline
+classifier = pipeline('text-classification', model="wcyat/bert-suicide-detection-hk")
+
+def classify_text(text):
+    # Get predictions
+    results = classifier(text)
+    # Extract and format the results
+    output = {result['label']: result['score'] for result in results}
+    return output
+
+import gradio as gr
+
+# Define Gradio interface
+iface = gr.Interface(
+    fn=classify_text,  # function to use for prediction
+    inputs="text",     # input type
+    outputs="label",   # output type
+    title="Text Classification with BERT",
+    description="Enter a sentence to classify whether it is suicidal."
+)
+
+# Launch the interface
+iface.launch()