alisamak commited on
Commit
8d126ec
·
verified ·
1 Parent(s): bed7954

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -37,11 +37,11 @@ def download_gaia_to_csv():
37
 
38
  login(token)
39
 
40
- # Load only Level 1 questions from the dev set
41
- dataset = load_dataset("gaia-benchmark/GAIA", "2023_level1", split="dev", trust_remote_code=True)
42
 
43
  # Save to CSV
44
- output_path = "gaia_dev_level1.csv"
45
  keys = dataset[0].keys()
46
 
47
  with open(output_path, "w", newline="", encoding="utf-8") as f:
@@ -50,7 +50,7 @@ def download_gaia_to_csv():
50
  for row in dataset:
51
  writer.writerow(row)
52
 
53
- return f"✅ GAIA Level 1 dev dataset saved to {output_path}"
54
  except Exception as e:
55
  return f"❌ Error: {e}"
56
 
 
37
 
38
  login(token)
39
 
40
+ # Load GAIA Level 1 validation set
41
+ dataset = load_dataset("gaia-benchmark/GAIA", "2023_level1", split="validation", trust_remote_code=True)
42
 
43
  # Save to CSV
44
+ output_path = "gaia_level1_validation.csv"
45
  keys = dataset[0].keys()
46
 
47
  with open(output_path, "w", newline="", encoding="utf-8") as f:
 
50
  for row in dataset:
51
  writer.writerow(row)
52
 
53
+ return f"✅ GAIA Level 1 validation set saved to {output_path}"
54
  except Exception as e:
55
  return f"❌ Error: {e}"
56