Spaces:
Sleeping
Sleeping
Commit
·
c5b95e6
1
Parent(s):
6618bc7
fix: load dataset
Browse files- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -22,7 +22,7 @@ def load_documents():
|
|
| 22 |
dataset_split = "basketball"
|
| 23 |
st.caption(f'Fetching "{dataset_name}" dataset')
|
| 24 |
# build the dataset
|
| 25 |
-
dataset = load_dataset(dataset_name, dataset_split)
|
| 26 |
docs_validation = get_unique_docs(dataset["validation"], unique_docs)
|
| 27 |
docs_train = get_unique_docs(dataset["train"], unique_docs)
|
| 28 |
docs_test = get_unique_docs(dataset["test"], unique_docs)
|
|
|
|
| 22 |
dataset_split = "basketball"
|
| 23 |
st.caption(f'Fetching "{dataset_name}" dataset')
|
| 24 |
# build the dataset
|
| 25 |
+
dataset = load_dataset(dataset_name, name=dataset_split)
|
| 26 |
docs_validation = get_unique_docs(dataset["validation"], unique_docs)
|
| 27 |
docs_train = get_unique_docs(dataset["train"], unique_docs)
|
| 28 |
docs_test = get_unique_docs(dataset["test"], unique_docs)
|
requirements.txt
CHANGED
|
@@ -2,7 +2,8 @@
|
|
| 2 |
# streamlit # commented to avoid miss instalation
|
| 3 |
|
| 4 |
# HuggingFace
|
| 5 |
-
datasets==2.
|
|
|
|
| 6 |
# Question Answering
|
| 7 |
haystack-ai==2.0.1
|
| 8 |
accelerate==0.29.2
|
|
|
|
| 2 |
# streamlit # commented to avoid miss instalation
|
| 3 |
|
| 4 |
# HuggingFace
|
| 5 |
+
datasets==3.2.0
|
| 6 |
+
huggingface_hub==0.27.0
|
| 7 |
# Question Answering
|
| 8 |
haystack-ai==2.0.1
|
| 9 |
accelerate==0.29.2
|