Update app.py

app.py

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