Spaces:
Sleeping
Sleeping
Ashmi Banerjee
commited on
Commit
·
330482c
1
Parent(s):
64656b5
updated dataset
Browse files
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]
|