Update app.py

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 from huggingface_hub import get_token
 
 from chatbot import get_retrieval_qa
-#from flagging import myHuggingFaceDatasetSaver as HuggingFaceDatasetSaver
-from gradio.flagging import HuggingFaceDatasetSaver
+from flagging import myHuggingFaceDatasetSaver as HuggingFaceDatasetSaver
+#from gradio.flagging import HuggingFaceDatasetSaver
 
 
 # get the html data and save it to a file
@@ -34,7 +34,8 @@ def answer_question(message, history, system):
     result = result.replace('"', "").strip()  # clean up the result
 
     # save the query and result to the dataset
-    hf_writer.flag(flag_data=[(query, result)])
+    print("flagging...")
+    hf_writer.flag(flag_data=[query, result])
     return result
 
 
@@ -98,6 +99,7 @@ with gr.ChatInterface(
     demo.load(lambda: download_html(url, filename))
     # This needs to be called prior to the first call to callback.flag()
     hf_writer.setup([demo.textbox, demo.chatbot], "flagged")
+    print("flagging setup done!")
     # like callback
     chatbot.like(on_like, None, None)