File size: 1,487 Bytes
324ecd2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | """Publish the prepared Space using the user's Modal-held HF write token.
The token is returned to this local process in memory only; it is never printed
or saved. Existing repositories are never overwritten by this script.
"""
import os
from pathlib import Path
import modal
app = modal.App('beamboot-publish')
@app.function(image=modal.Image.debian_slim(python_version='3.12'), secrets=[modal.Secret.from_name('steerability-submission-write')])
def get_write_token():
token = os.environ.get('HUGGINGFACE_TOKEN') or os.environ.get('HF_TOKEN')
if not token:
raise RuntimeError('No HF write token in configured secret')
return token
@app.local_entrypoint()
def main():
from huggingface_hub import HfApi
token = get_write_token.remote()
api = HfApi(token=token)
name = api.whoami()['name']
repo_id = f'{name}/beam-boot'
if not api.repo_exists(repo_id=repo_id, repo_type='space'):
api.create_repo(repo_id=repo_id, repo_type='space', space_sdk='static', private=False)
print(f'Uploading to Space created for this task: https://huggingface.co/spaces/{repo_id}', flush=True)
api.upload_folder(repo_id=repo_id, repo_type='space', folder_path=str(Path('static-space').resolve()),
commit_message='Add Super Beam: Beam plays a game that runs inside Beam',
ignore_patterns=['**/__pycache__/**', '**/*.pyc'])
print(f'Uploaded files: https://huggingface.co/spaces/{repo_id}', flush=True)
|