Update main.py

main.py

@@ -16,13 +16,17 @@ step_threshold = 5
 HF_DATASET_ID = "agents-course/unit4-students-scores"
 # Ensure you have write access to this dataset repository on Hugging Face
 # and are logged in via `huggingface-cli login` or have HF_TOKEN env var set.
-
+    # Prepare data structures for the API
+questions_for_api: List[Dict[str, str]] = []
+ground_truth_answers: Dict[str, str] = {}
 # --- Logging Setup ---
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 filtered_dataset=None
 def load_dataset():
     global filtered_dataset
+    global questions_for_api
+    global ground_truth_answers
     tempo_filtered=[]
     dataset=load_dataset("gaia-benchmark/GAIA","2023_level1",trust_remote_code=True)
     for question in dataset['validation']:
@@ -53,9 +57,7 @@ def load_dataset():
                     print("------------------------------------------------------------------")
     filtered_dataset=tempo_filtered
 
-    # Prepare data structures for the API
-    questions_for_api: List[Dict[str, str]] = []
-    ground_truth_answers: Dict[str, str] = {}
+
     
     for item in filtered_dataset:
         task_id = item.get('task_id')