Update app.py

app.py

@@ -13,7 +13,7 @@ class CustomFlaggingCallback(gr.FlaggingCallback):
     def __init__(self, hf_token, dataset_name):
         self.hf_writer = gr.HuggingFaceDatasetSaver(hf_token, dataset_name)
 
-    def setup(self, interface):
+    def setup(self, interface, interface_state):
         pass
 
     def flag(self, flagged_data):