Update helper.py

helper.py

@@ -1,15 +1,12 @@
 import pandas as pd
 
-#FILE_PATH = "https://huggingface.co/datasets/gaia-benchmark/GAIA/blob/main/2023/validation/"
-
-def get_questions():
+def get_questions(level):
     df = pd.read_json("gaia.jsonl", lines = True)
+    level_df = df[df["Level"] == level]
 
     result = []
     
-    for index, row in df.iterrows():
+    for index, row in level_df.iterrows():
         result.append([row["Level"], row["Question"], row["file_name"], row["Final answer"]])
 
-    sorted_result = sorted(result, key = lambda x: x[0])
-    
-    return sorted_result
+    return result