Update app.py

app.py

@@ -8,10 +8,6 @@ import json
 # Set the OpenAI API key
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-# Set up the Hugging Face Dataset Saver
-HF_TOKEN = os.getenv("HF_TOKEN")
-hf_writer = CustomHuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
-
 # Define the CustomHuggingFaceDatasetSaver class
 class CustomHuggingFaceDatasetSaver(gr.FlaggingCallback):
     def setup(self):
@@ -34,6 +30,10 @@ class CustomHuggingFaceDatasetSaver(gr.FlaggingCallback):
                 writer = csv.writer(f)
                 writer.writerow([input_data, output_data, flag])
 
+# Set up the Hugging Face Dataset Saver
+HF_TOKEN = os.getenv("HF_TOKEN")
+hf_writer = CustomHuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
+
 # Define the authentication function
 def check_auth(username, password):
     # Get the credentials from the environment variable