Update app.py

app.py

@@ -9,7 +9,7 @@ import json
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
 # Define the CustomHuggingFaceDatasetSaver class
-class CustomHuggingFaceDatasetSaver(gr.FlaggingCallback):
+class HuggingFaceDatasetSaver(gr.FlaggingCallback):
     def setup(self):
         pass
 
@@ -32,7 +32,7 @@ class CustomHuggingFaceDatasetSaver(gr.FlaggingCallback):
 
 # Set up the Hugging Face Dataset Saver
 HF_TOKEN = os.getenv("HF_TOKEN")
-hf_writer = gr.CustomHuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
 
 # Define the authentication function
 def check_auth(username, password):