File size: 1,466 Bytes
a33458e |
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 |
#!/usr/bin/env python
"""
Register this app with Hugging Face Spaces SDK.
This file is used for deploying the app to Hugging Face Spaces.
"""
import os
import subprocess
import sys
from huggingface_hub import SpaceHardware, SpaceStage, SpaceSDK
def create_space():
"""Create or update a Hugging Face Space."""
# Get the Space name or use a default
space_name = os.environ.get("SPACE_NAME", "personal-rag-assistant")
owner = os.environ.get("HF_USERNAME")
if not owner:
print("Please set the HF_USERNAME environment variable to your Hugging Face username.")
sys.exit(1)
# Initialize the SDK
sdk = SpaceSDK(
space_id=f"{owner}/{space_name}",
token=os.environ.get("HF_TOKEN")
)
# Check if space exists, if not create it
try:
space_info = sdk.get_space_runtime()
print(f"Space {owner}/{space_name} exists.")
exists = True
except Exception:
exists = False
# Create or update the space
if not exists:
print(f"Creating new space: {owner}/{space_name}")
sdk.create_space(
space_hardware=SpaceHardware.CPU_BASIC,
space_storage=1,
space_sleep_time=3600, # 1 hour of inactivity before sleep
space_stage=SpaceStage.RUNNING,
)
print(f"Space URL: https://huggingface.co/spaces/{owner}/{space_name}")
return sdk
if __name__ == "__main__":
create_space() |