bstraehle commited on
Commit
cfe90dc
·
verified ·
1 Parent(s): aaa663a

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +4 -2
utils.py CHANGED
@@ -1,10 +1,12 @@
1
- import os
2
  import pandas as pd
3
  from docx import Document
4
  from pptx import Presentation
5
 
6
  def get_questions(file_path, level):
7
- df = pd.read_json(file_path, lines=True)
 
 
8
 
9
  if level > 0:
10
  df = df[df["Level"] == level]
 
1
+ import datasets, os
2
  import pandas as pd
3
  from docx import Document
4
  from pptx import Presentation
5
 
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")["validation"]
10
 
11
  if level > 0:
12
  df = df[df["Level"] == level]