dawalina's picture
Added hf_token file
8cb8f25
Raw
History Blame Contribute Delete
311 Bytes
import os
from huggingface_hub import get_token
def resolve_hf_token() -> str | None:
"""Return HF API token from env or huggingface-cli cache."""
for key in ("HF_TOKEN", "HUGGING_FACE_HUB_TOKEN"):
value = os.environ.get(key)
if value:
return value
return get_token()