Update app.py

app.py

@@ -8,13 +8,14 @@ pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-b
 
 
 def detect_suicide_comment(input):
-    return pipe(input)[0]
+    result = pipe(input)[0]
+    return {result['label']: result['score']}
 
 
 iface = gr.Interface(
     fn=detect_suicide_comment,
     inputs="text",
-    outputs="text",
+    outputs=gr.Label(num_top_classes=1, type="confidences"),
     allow_flagging="manual",
     flagging_options=['Correct Classification', 'Incorrect Classification'],
     flagging_callback=hf_writer,