Update app.py

app.py

@@ -1,7 +1,9 @@
+import os
 from transformers import pipeline
 import gradio as gr
 
 
+hf_writer = gr.HuggingFaceDatasetSaver(os.getenv('HF_API_TOKEN'), "suicide-comment-es")
 pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es')
 
 
@@ -10,6 +12,11 @@ def detect_suicide_comment(input):
 
 
 iface = gr.Interface(
-    fn=detect_suicide_comment, inputs="text", outputs="text", allow_flagging="auto"
+    fn=detect_suicide_comment,
+    inputs="text",
+    outputs="text",
+    allow_flagging="manual",
+    flagging_options='Incorrect classification',
+    flagging_callback=hf_writer
 )
 iface.launch()