File size: 2,000 Bytes
6c3d778 | 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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 | from __future__ import annotations
import os
from pathlib import Path
from typing import Mapping
from huggingface_hub import HfApi
SPACE_NAME = "amongus-openenv"
def main() -> None:
token = os.environ.get("HF_TOKEN")
if not token:
raise SystemExit("Set HF_TOKEN in the environment before running this script.")
api = HfApi(token=token)
repo_id = resolve_repo_id(api=api)
api.create_repo(
repo_id=repo_id,
repo_type="space",
space_sdk="gradio",
private=False,
exist_ok=True,
)
api.upload_folder(
folder_path=str(Path(__file__).resolve().parents[1]),
repo_id=repo_id,
repo_type="space",
ignore_patterns=[
".git/*",
".pytest_cache/*",
"__pycache__/*",
"*.pyc",
],
)
print(f"https://huggingface.co/spaces/{repo_id}")
def resolve_repo_id(
environ: Mapping[str, str] | None = None,
api: HfApi | None = None,
) -> str:
environ = os.environ if environ is None else environ
space_id = environ.get("HF_SPACE_ID")
if space_id:
if "/" not in space_id:
raise SystemExit("HF_SPACE_ID must be in the form username_or_org/space_name.")
return space_id
username = environ.get("HF_USERNAME")
space_name = environ.get("HF_SPACE_NAME", SPACE_NAME)
if username:
return f"{username}/{space_name}"
if api is not None:
try:
return f"{api.whoami(cache=True)['name']}/{space_name}"
except Exception as exc:
raise SystemExit(
"Could not infer your Hugging Face username without hitting a "
"rate-limited endpoint. Set HF_SPACE_ID=username_or_org/space_name "
"or HF_USERNAME=username_or_org, then rerun."
) from exc
raise SystemExit(
"Set HF_SPACE_ID=username_or_org/space_name or HF_USERNAME=username_or_org."
)
if __name__ == "__main__":
main()
|