350016z commited on
Commit
a82b5c3
·
verified ·
1 Parent(s): 8ae819b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ scheduler = CommitScheduler(
18
  )
19
 
20
  def download_dataset_file(dataset_id, file_name, local_dir):
21
- dataset = load_dataset(dataset_id)
22
  local_file_path = os.path.join(local_dir, file_name)
23
 
24
  df = pd.DataFrame(dataset)
 
18
  )
19
 
20
  def download_dataset_file(dataset_id, file_name, local_dir):
21
+ dataset = load_dataset(dataset_id, split="test")
22
  local_file_path = os.path.join(local_dir, file_name)
23
 
24
  df = pd.DataFrame(dataset)