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]
|