Update app.py

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 import pandas as pd
 from retrieval import *
+import os
+
+HF_TOKEN = os.getenv('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')
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "chitti_data")
 
 df = pd.read_csv("Responses.csv")
 text = list(df["text"].values)
@@ -18,7 +22,7 @@ demo = gr.Interface(
     outputs=["text","text"],
     allow_flagging = "manual",
     flagging_options = ["1","2","None"],
-    flagging_dir = "Questions"
+    flagging_callback=hf_writer
 )
 demo.launch()