Fixed things

app.py

@@ -1,13 +1,18 @@
+import spaces
+
 import gradio as gr
+import torch
 from transformers import pipeline
 
+
 pipe = pipeline(
     "text-classification",
     model="SHK4K/suicide-roberta",
-    top_k=None,
+    device=0 if torch.cuda.is_available() else -1,
 )
 
 
+@spaces.GPU
 def is_safe(text):
     if not text or not text.strip():
         return {
@@ -19,6 +24,7 @@ def is_safe(text):
         text,
         truncation=True,
         max_length=512,
+        top_k=None,
     )[0]
 
     return {
@@ -44,7 +50,7 @@ demo = gr.Interface(
     title="Suicide Risk Detector",
 
     description=(
-        "A research model that detects potential "
+        "A research NLP model that detects potential "
         "suicide-risk signals in text. "
         "This is not a medical diagnosis."
     ),

requirements.txt

@@ -1,3 +1,5 @@
 gradio
 transformers
 torch
+spaces
+sentencepiece

requirments.txt

@@ -1,3 +0,0 @@
-gradio
-transformers
-torch