Update app.py

app.py

@@ -14,11 +14,14 @@ class CustomHuggingFaceDatasetSaver(gr.FlaggingCallback):
         self.hf_token = hf_token
         self.dataset_name = dataset_name
 
-    def on_flag(self, sample):
+    def flag(self, sample):
         if len(sample["output"]) > 1000:
             sample["output"] = sample["output"][:1000]  # Truncate output to 1000 characters
 
-        super().on_flag(sample)
+        super().flag(sample)
+
+    def setup(self):
+        pass
 
 # Set up flagging callback function
 HF_TOKEN = os.getenv("HF_TOKEN")