Update app.py

app.py

@@ -10,10 +10,11 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 
 # Set up the Hugging Face Dataset Saver
 HF_TOKEN = os.getenv("HF_TOKEN")
+DATASET_NAME = "CannaTech/Flagged"
 
 class CustomFlaggingCallback:
     def __init__(self, hf_token, dataset_name):
-        self.hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
+        self.hf_writer = gr.HuggingFaceDatasetSaver(hf_token, dataset_name)
 
     def __call__(self, input_data, output_data):
         # Custom logic to handle flagged data
@@ -22,7 +23,7 @@ class CustomFlaggingCallback:
         self.hf_writer.flush()
 
 # Create an instance of the custom flagging callback
-custom_callback = CustomFlaggingCallback(HF_TOKEN, "CannaTech/Flagged")
+custom_callback = CustomFlaggingCallback(HF_TOKEN, DATASET_NAME)
 
         
 # Define the authentication function