Ashmi Banerjee commited on
Commit
330482c
·
1 Parent(s): 64656b5

updated dataset

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ def load_data():
94
  data_files = os.getenv("LLAMA_DATA_FILES")
95
  print("data_files", data_files)
96
  HF_TOKEN = os.getenv("HF_TOKEN")
97
- dataset = load_dataset(repo_name, token=True, data_files=data_files)
98
  dataset.set_format(type='pandas') ## converting it into pandas
99
  df = dataset["train"][:]
100
  return df[:5]
 
94
  data_files = os.getenv("LLAMA_DATA_FILES")
95
  print("data_files", data_files)
96
  HF_TOKEN = os.getenv("HF_TOKEN")
97
+ dataset = load_dataset(repo_name, token=True, data_files=data_files, revision="main")
98
  dataset.set_format(type='pandas') ## converting it into pandas
99
  df = dataset["train"][:]
100
  return df[:5]