Update utils.py
Browse files
utils.py
CHANGED
|
@@ -6,7 +6,7 @@ from pptx import Presentation
|
|
| 6 |
def get_questions(file_path, level):
|
| 7 |
#df = pd.read_json(file_path, lines=True)
|
| 8 |
|
| 9 |
-
df = datasets.load_dataset("gaia-benchmark/GAIA", "2023_all", token=True)["validation"]
|
| 10 |
|
| 11 |
if level > 0:
|
| 12 |
df = df[df["Level"] == level]
|
|
|
|
| 6 |
def get_questions(file_path, level):
|
| 7 |
#df = pd.read_json(file_path, lines=True)
|
| 8 |
|
| 9 |
+
df = datasets.load_dataset("gaia-benchmark/GAIA", "2023_all", token=True, trust_remote_code=True)["validation"]
|
| 10 |
|
| 11 |
if level > 0:
|
| 12 |
df = df[df["Level"] == level]
|