dreambooth / main.py
abhishek's picture
abhishek HF staff
first commit
acc62e6
import os
import glob
from huggingface_hub import snapshot_download
REPO_ID = os.environ.get("REPO_ID")
TOKEN = os.environ.get("TOKEN")
folder = snapshot_download(
repo_id=REPO_ID,
revision="main",
allow_regex="*.py",
repo_type="space",
use_auth_token=TOKEN,
)
# copy all *.py files in folder to .
for file in glob.glob(os.path.join(folder, "*.py")):
os.system(f"cp {file} .")
from app import demo
demo.launch()