Update app.py

app.py

@@ -1,10 +1,8 @@
-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')
-hf_writer = gr.HuggingFaceDatasetSaver(os.environ['HF_API_TOKEN'], "text-classification-mistakes")
 
 
 def detect_suicide_comment(input):
@@ -12,6 +10,6 @@ def detect_suicide_comment(input):
 
 
 iface = gr.Interface(
-    fn=detect_suicide_comment, inputs="text", outputs="text", allow_flagging="manual", flagging_callback=hf_writer
+    fn=detect_suicide_comment, inputs="text", outputs="text"
 )
 iface.launch()