Spaces:
Runtime error
Runtime error
Commit
·
e96101e
1
Parent(s):
e6064e2
chore: Update data_loader.py with Hugging Face Hub integration
Browse files- data_loader.py +2 -1
data_loader.py
CHANGED
|
@@ -20,6 +20,7 @@ assert (
|
|
| 20 |
logger = logging.get_logger(__name__)
|
| 21 |
|
| 22 |
api = HfApi(token=HF_TOKEN)
|
|
|
|
| 23 |
|
| 24 |
|
| 25 |
def has_card_data(dataset):
|
|
@@ -95,7 +96,7 @@ def refresh_data() -> List[Dict[str, Any]]:
|
|
| 95 |
return df.to_dict(orient="records")
|
| 96 |
|
| 97 |
# List all datasets
|
| 98 |
-
datasets = list(api.list_datasets(limit=
|
| 99 |
|
| 100 |
# Filter datasets with dataset info
|
| 101 |
datasets = [
|
|
|
|
| 20 |
logger = logging.get_logger(__name__)
|
| 21 |
|
| 22 |
api = HfApi(token=HF_TOKEN)
|
| 23 |
+
MAX_DATASETS = 1000
|
| 24 |
|
| 25 |
|
| 26 |
def has_card_data(dataset):
|
|
|
|
| 96 |
return df.to_dict(orient="records")
|
| 97 |
|
| 98 |
# List all datasets
|
| 99 |
+
datasets = list(api.list_datasets(limit=MAX_DATASETS, full=True))
|
| 100 |
|
| 101 |
# Filter datasets with dataset info
|
| 102 |
datasets = [
|