Update app.py

app.py

@@ -98,6 +98,10 @@ def predict(input):
   outputs = model.generate(input_ids, max_length=512)
   outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
   return outputs
+  
+HF_TOKEN = os.getenv('spanish-nahuatl-flagging')
+
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "spanish-nahuatl-flagging")
 
 gr.Interface(
    fn=predict,
@@ -122,5 +126,5 @@ gr.Interface(
    article=article,
    allow_flagging="manual",
    flagging_options=["right translation", "wrong translation", "error", "other"],
-   flagging_dir="logs"
+   flagging_callback=hf_writer,
    ).launch(enable_queue=True)