File size: 832 Bytes
ae52029 | 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 | #!/bin/bash
set -e
if [ ! -f "$MODEL_CHECKPOINT" ]; then
echo "[startup] Checkpoint not found at $MODEL_CHECKPOINT"
if [ -z "$HF_MODEL_REPO" ]; then
echo "[startup] ERROR: set HF_MODEL_REPO env var (e.g. nitya2405/sangeet-hindustani-lm)"
exit 1
fi
echo "[startup] Downloading checkpoint from $HF_MODEL_REPO ..."
python - <<'PYEOF'
import os, shutil
from huggingface_hub import hf_hub_download
dst = os.environ["MODEL_CHECKPOINT"]
path = hf_hub_download(
repo_id=os.environ["HF_MODEL_REPO"],
filename=os.environ.get("HF_CKPT_FILE", "checkpoint.pt"),
token=os.environ.get("HF_TOKEN"),
local_dir="/tmp/hf_dl",
local_dir_use_symlinks=False,
)
shutil.copy(path, dst)
print(f"[startup] Checkpoint saved to {dst}")
PYEOF
fi
exec uvicorn backend.main:app --host 0.0.0.0 --port 7860
|