Update app.py

app.py

@@ -7,6 +7,10 @@ import json
 # Set the OpenAI API key
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
+# Set up flagging callback function
+HF_TOKEN = os.getenv("HF_TOKEN")
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
+
 # Define the authentication function
 def check_auth(username, password):
     # Get the credentials from the environment variable
@@ -130,6 +134,7 @@ iface = gr.Interface(
     thumbnail="https://assets.bigcartel.com/theme_images/101321509/IMG_6002.png",  # Thumbnail image URL
     favicon_path="https://assets.bigcartel.com/theme_images/101321509/IMG_6002.png",  # Favicon image URL
     allow_flagging="manual",
+    flagging_callback=hf_writer
    
 )