Spaces:
Paused
Paused
| #!/usr/bin/env python3 | |
| """Deploy DepthWizard to Hugging Face Spaces using huggingface_hub.""" | |
| from __future__ import annotations | |
| import argparse | |
| import sys | |
| from pathlib import Path | |
| ROOT = Path(__file__).resolve().parents[1] | |
| IGNORE_PATTERNS = [ | |
| "venv/*", | |
| ".venv/*", | |
| "*.pyc", | |
| "__pycache__/*", | |
| ".DS_Store", | |
| ".git/*", | |
| "data/*", | |
| "*.h5", | |
| "outputs/stage12/checkpoints/checkpoint_*.pt", | |
| "outputs/stage12/checkpoints/experiment_A_*.pt", | |
| "outputs/stage12/checkpoints/smoke_test_*.pt", | |
| "outputs/final/runs/*", | |
| "tests/*", | |
| ] | |
| def main(): | |
| parser = argparse.ArgumentParser(description=__doc__) | |
| parser.add_argument("--repo-id", help="Hugging Face Space repo ID (e.g. 'username/depthwizard')") | |
| parser.add_argument("--token", help="Hugging Face API write token (optional if logged in via `huggingface-cli login`)") | |
| parser.add_argument("--private", action="store_true", help="Create space as private (default is public)") | |
| args = parser.parse_args() | |
| try: | |
| from huggingface_hub import HfApi, login | |
| except ImportError: | |
| print("Error: huggingface_hub is not installed in the active environment.") | |
| print("Install it with: ./venv/bin/python -m pip install huggingface_hub") | |
| sys.exit(1) | |
| api = HfApi(token=args.token) | |
| # Determine token and user | |
| user_info = None | |
| try: | |
| user_info = api.whoami() | |
| username = user_info["name"] | |
| print(f"Authenticated as Hugging Face user: {username}") | |
| except Exception as exc: | |
| if not args.token: | |
| print("Authentication required. Please provide --token or run `huggingface-cli login`.") | |
| print("You can get a free token with write access at: https://huggingface.co/settings/tokens") | |
| token_input = input("Enter your Hugging Face write token: ").strip() | |
| if not token_input: | |
| sys.exit(1) | |
| api = HfApi(token=token_input) | |
| user_info = api.whoami() | |
| username = user_info["name"] | |
| else: | |
| print(f"Error authenticating with provided token: {exc}") | |
| sys.exit(1) | |
| repo_id = args.repo_id | |
| if not repo_id: | |
| default_repo = f"{username}/depthwizard" | |
| user_choice = input(f"Enter Space name [default: {default_repo}]: ").strip() | |
| repo_id = user_choice if user_choice else default_repo | |
| if "/" not in repo_id: | |
| repo_id = f"{username}/{repo_id}" | |
| print(f"\nTarget Space: https://huggingface.co/spaces/{repo_id}") | |
| print("Creating/verifying Space on Hugging Face...") | |
| try: | |
| api.create_repo( | |
| repo_id=repo_id, | |
| repo_type="space", | |
| space_sdk="docker", | |
| private=args.private, | |
| exist_ok=True, | |
| ) | |
| print("Space is ready for deployment.") | |
| except Exception as exc: | |
| print(f"Notice: {exc}") | |
| print("\nUploading project files to Hugging Face Spaces (this may take a couple minutes for model weights)...") | |
| try: | |
| commit_info = api.upload_folder( | |
| folder_path=str(ROOT), | |
| repo_id=repo_id, | |
| repo_type="space", | |
| ignore_patterns=IGNORE_PATTERNS, | |
| commit_message="Deploy DepthWizard frontend and backend", | |
| ) | |
| print("\nUpload complete!") | |
| print(f"Live Space URL: https://huggingface.co/spaces/{repo_id}") | |
| print("Hugging Face will now build the Docker container and start DepthWizard.") | |
| print("You can track build logs directly at the link above.") | |
| except Exception as exc: | |
| print(f"Upload failed: {exc}") | |
| sys.exit(1) | |
| if __name__ == "__main__": | |
| main() | |