Spaces:
Runtime error
Runtime error
fix: ensure Hugging Face API token is used for dataset repository initialization
Browse files- utils/hf_logger.py +1 -1
utils/hf_logger.py
CHANGED
|
@@ -37,7 +37,7 @@ def _pil_to_base64(image: Image.Image) -> str:
|
|
| 37 |
# as we are no longer appending to a datasets.Dataset object directly in memory
|
| 38 |
def initialize_dataset_repo():
|
| 39 |
"""Initializes or ensures the Hugging Face dataset repository exists."""
|
| 40 |
-
api = HfApi()
|
| 41 |
try:
|
| 42 |
api.repo_info(repo_id=HF_DATASET_NAME, repo_type="dataset")
|
| 43 |
logger.info(f"Hugging Face dataset repository already exists: {HF_DATASET_NAME}")
|
|
|
|
| 37 |
# as we are no longer appending to a datasets.Dataset object directly in memory
|
| 38 |
def initialize_dataset_repo():
|
| 39 |
"""Initializes or ensures the Hugging Face dataset repository exists."""
|
| 40 |
+
api = HfApi(token=os.getenv("HF_TOKEN"))
|
| 41 |
try:
|
| 42 |
api.repo_info(repo_id=HF_DATASET_NAME, repo_type="dataset")
|
| 43 |
logger.info(f"Hugging Face dataset repository already exists: {HF_DATASET_NAME}")
|