Update app.py

app.py

@@ -5,6 +5,7 @@ import huggingface_hub
 import os
 
 huggingface_hub.login(token=os.environ["HF_TOKEN"])
+hf_writer = gr.HuggingFaceDatasetSaver(os.environ["HF_WRITE_TOKEN"], "fava-flagged-demo")
 
 # Fava prompt
 INPUT = "Read the following references:\n{evidence}\nPlease identify all the errors in the following text using the information in the references provided and suggest edits if necessary:\n[Text] {output}\n[Edited] "
@@ -44,5 +45,5 @@ if __name__ == "__main__":
                  "Banff National Park is Canada's oldest national park, established in 1885 as Rocky Mountains Park. Located in Alberta's Rocky Mountains, 110–180 kilometres (68–112 mi) west of Calgary, Banff encompasses 6,641 square kilometres (2,564 sq mi) of mountainous terrain."]]
     demo = gradio.Interface(fn=result, inputs=["text", "text"], outputs="html", title="Fine-grained Hallucination Detection and Editing (FAVA)", 
                             description="Given a passage and a reference, FAVA will detect and edit any hallucinations present in the passage.", article=article,
-                           examples=examples, allow_flagging="manual", flagging_options=["wrong detection", "wrong editing", "both wrong", "other"], flagging_dir="flags")
+                           examples=examples, allow_flagging="manual", flagging_options=["wrong detection", "wrong edit", "both wrong", "other"], flagging_callback=hf_writer)
     demo.launch(share=True)