LPX
commited on
Commit
·
f69af75
1
Parent(s):
489bc21
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}")
|