Update app.py

app.py

@@ -9,7 +9,8 @@ pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-b
 
 def detect_suicide_comment(input):
     result = pipe(input)[0]
-    return {result['label']: result['score']}
+    # return {result['label']: result['score']}
+    return result
 
 
 iface = gr.Interface(