Add application file

app.py

@@ -0,0 +1,33 @@
+import gradio as gr
+from transformers import pipeline
+
+pipe = pipeline(
+    "text-classification",
+    model="SHK4K/suicide-roberta",
+)
+
+def is_safe(text,exp=None):
+    inputs = tokenizer(
+    text,
+    padding=True,
+    truncation=True,
+    max_length=512,
+    return_tensors="pt",
+)
+    with torch.no_grad():
+        logits = model(**inputs).logits
+        predicted_label = logits.argmax(dim=-1).item()
+        confidence = logits.softmax(dim=-1)[0][predicted_label].item()
+
+    label_map = {0: "safe", 1: "unsafe"}
+
+    print(f"Prediction: {label_map[predicted_label]} (confidence: {confidence:.3f}) {f"Expected: {label_map[exp]}" if exp is not None else ""}")
+    return label_map[predicted_label]
+
+demo = gr.Interface(
+    fn=is_safe,
+    inputs="textbox",
+    outputs="label",
+)
+
+demo.launch(share=True)