Update app.py

app.py

@@ -12,10 +12,13 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 HF_TOKEN = os.getenv("HF_TOKEN")
 DATASET_NAME = "CannaTech/Flagged"
 
-class CustomFlaggingCallback:
+class CustomFlaggingCallback(gr.FlaggingCallback):
     def __init__(self, hf_token, dataset_name):
         self.hf_writer = gr.HuggingFaceDatasetSaver(hf_token, dataset_name)
 
+    def setup(self, interface):
+        pass
+
     def __call__(self, input_data, output_data):
         # Custom logic to handle flagged data
         # Example: Save flagged data to Hugging Face Dataset without writing the output to JSON file