Add application file flag

app.py

@@ -1,13 +1,15 @@
+import os
 from transformers import pipeline
 import gradio as gr
 
 
-PIPE = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es')
+pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es')
+hf_writer = gr.HuggingFaceDatasetSaver(os.environ['HF_API_TOKEN'], "text-classification-mistakes")
 
 
-def detect_suicide_comment(comentario):
-    return PIPE(comentario)
+def detect_suicide_comment(comentario="Quiero acabar con todo. No merece la pena vivir."):
+    return pipe(comentario)
 
 
-iface = gr.Interface(fn=detect_suicide_comment, inputs="text", outputs="text")
+iface = gr.Interface(fn=detect_suicide_comment, inputs="text", outputs="text", flagging_callback=hf_writer)
 iface.launch()