|
|
|
|
|
from pathlib import Path |
|
from dotenv import dotenv_values |
|
from huggingface_hub import HfApi |
|
|
|
|
|
dotenv_path = Path(__file__).resolve().parent.parent / '.env' |
|
print(dotenv_path) |
|
|
|
|
|
env_vars = dotenv_values(dotenv_path) |
|
|
|
|
|
SPACE_NAME: str = env_vars.pop('SPACE_NAME', '') |
|
HF_TOKEN: str = env_vars['HF_TOKEN'] |
|
if SPACE_NAME: |
|
api = HfApi(token = HF_TOKEN) |
|
for k,v in env_vars.items(): |
|
api.add_space_secret( |
|
repo_id=SPACE_NAME, |
|
key=k, |
|
value=v, |
|
) |
|
else: |
|
print('You should set "SPACE_NAME" in .env files') |
|
|
|
print('Update Keys Successfully!') |
|
|
