LPX commited on
Commit
f69af75
·
1 Parent(s): 489bc21

fix: ensure Hugging Face API token is used for dataset repository initialization

Browse files
Files changed (1) hide show
  1. 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}")