Update app.py

app.py

@@ -8,24 +8,9 @@ import json
 # Set the OpenAI API key
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-class CustomHuggingFaceDatasetSaver(gr.FlaggingCallback):
-    def __init__(self, hf_token, dataset_name):
-        super().__init__()
-        self.hf_token = hf_token
-        self.dataset_name = dataset_name
-
-    def flag(self, sample):
-        if len(sample["output"]) > 1000:
-            sample["output"] = sample["output"][:1000]  # Truncate output to 1000 characters
-
-        super().flag(sample)
-
-    def setup(self):
-        pass
-
 # Set up flagging callback function
 HF_TOKEN = os.getenv("HF_TOKEN")
-hf_writer = CustomHuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
 
 # Define the authentication function
 def check_auth(username, password):