Update app.py

app.py

@@ -14,7 +14,7 @@ def detect_suicide_comment(input):
 iface = gr.Interface(
     fn=detect_suicide_comment,
     inputs="text",
-    outputs="text",
+    outputs=gr.Label(num_top_classes=1),
     allow_flagging="manual",
     flagging_options=['Correct Classification', 'Incorrect Classification'],
     flagging_callback=hf_writer,