Spaces:
Runtime error
Runtime error
File size: 2,668 Bytes
57b35b1 |
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 |
import datetime
from huggingface_hub import create_repo, upload_folder, add_space_secret, whoami
from fastcore.utils import *
from fastcore.script import *
def _mk_docker(python_ver):
fn = Path('Dockerfile')
if fn.exists(): return
packages = Path('packages.txt')
pkg_line = ''
reqs = Path('requirements.txt')
if not reqs.exists(): reqs.write_text('python-fasthtml\nfasthtml-hf\n')
req_line = f'RUN pip install --no-cache-dir -r requirements.txt'
if packages.exists():
pkglist = ' '.join(packages.readlines())
pkg_line = f'RUN apt-get update -y && apt-get install -y {pkglist}'
cts = f"""FROM python:{python_ver}
WORKDIR /code
COPY --link --chown=1000 . .
RUN mkdir -p /tmp/cache/
RUN chmod a+rwx -R /tmp/cache/
ENV HF_HUB_CACHE=HF_HOME
{req_line}
{pkg_line}
ENV PYTHONUNBUFFERED=1 PORT=7860
CMD ["python", "main.py"]
"""
fn.write_text(cts)
def _mk_README(space_id, termination_grace_period):
fn = Path('README.md')
if fn.exists(): return
cts = f"""
---
title: {space_id}
emoji: 🚀
colorFrom: purple
colorTo: red
sdk: docker
app_file: app.py
pinned: false
termination_grace_period: {termination_grace_period}
---
"""
fn.write_text(cts)
@call_parse
def deploy(
space_id:str, # ID of the space to upload to
token:str=None, # Hugging Face token for authentication
python_ver:str='3.10', # Version of python to use
upload:bool_arg=True, # Set to `false` to skip uploading files
private:bool_arg=False,
termination_grace_period:str="2m"): # Make the repository private
"Upload current directory to Hugging Face Spaces"
if not token: token=os.getenv('HF_TOKEN')
if not token: return print('No token available')
if "/" not in space_id: space_id = f"{whoami(token)['name']}/{space_id}"
_mk_docker(python_ver)
_mk_README(space_id, termination_grace_period)
private = bool(private) # `private` can be 0,1 or False. As `create_repo` expects private to be True/False we cast it.
url = create_repo(space_id, token=token, repo_type='space',
space_sdk="docker", private=private, exist_ok=True)
if not upload: return print('Repo created; upload skipped')
upload_folder(folder_path=Path("."),
repo_id=space_id, repo_type='space',
ignore_patterns=['__pycache__/*', '.sesskey', 'deploy_hf.py', 'data/*'],
commit_message=f"deploy at {datetime.datetime.now()}",
token=token)
add_space_secret(space_id, token=token, key="HF_TOKEN", value=token)
print(f"Deployed space at {url}")
|