Update app.py

flagging with dir

app.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer
 from huggingface_hub import snapshot_download
 from codeexecutor import get_majority_vote
 import re
-
+import os
 # Define the model and tokenizer loading
 model_prompt = "Explain and solve the following mathematical problem step by step, showing all work: "
 tokenizer = AutoTokenizer.from_pretrained("AI-MO/NuminaMath-7B-TIR")
@@ -165,6 +165,13 @@ custom_css = """
     }
 """
 
+# Define the directory path
+flagging_dir = "./flagged_data"
+
+# Create the directory if it doesn't exist
+if not os.path.exists(flagging_dir):
+    os.makedirs(flagging_dir)
+
 # Gradio app setup with flagging
 interface = gr.Interface(
     fn=gradio_interface,
@@ -179,7 +186,7 @@ interface = gr.Interface(
     title="🔢 Math Question Solver",
     description="Enter a math question to get the model's majority-voted answer and steps to solve the problem.",
     css=custom_css,  # Apply custom CSS
-    flagging_dir="flagged_data",  # Directory to save flagged data
+    flagging_dir=flagging_dir,  # Directory to save flagged data
     allow_flagging="auto"  # Allow users to auto flag data
 )