Update app.py

app.py

@@ -3,6 +3,9 @@ from sentence_transformers import SentenceTransformer, util
 import openai
 import os
 
+HF_TOKEN = os.getenv('HF_TOKEN')
+hf_writer =gr.HuggingFaceDatasetSaver(HF_TOKEN, "RotBot_Flags")
+
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 # Initialize paths and model identifiers for easy configuration and maintenance
@@ -156,8 +159,9 @@ with gr.Blocks(theme='gradio/soft') as demo:
             ["What is the definition of gyatt?"]
         ],
         allow_flagging="manual",
-        flagging_dir = "flag",
-        flagging_options=["Ambiguous", "Wrong", "Other"]
+        # flagging_dir = "flag",
+        flagging_options=["Ambiguous", "Wrong", "Other"],
+        flagging_callback=hf_writer
 
     )