21 lines
562 B
Bash
21 lines
562 B
Bash
|
|
#!/usr/bin/env bash
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
MODEL_PATH="${ORMBG_MODEL_PATH:-/root/.ormbg/ormbg.pth}"
|
||
|
|
MODEL_DIR="$(dirname "$MODEL_PATH")"
|
||
|
|
|
||
|
|
if [ ! -f "$MODEL_PATH" ]; then
|
||
|
|
echo "[bgbye] ORMBG model not found at $MODEL_PATH - running setup.sh to fetch it..."
|
||
|
|
mkdir -p "$MODEL_DIR"
|
||
|
|
if [ -x /app/server/setup.sh ]; then
|
||
|
|
bash /app/server/setup.sh
|
||
|
|
else
|
||
|
|
echo "[bgbye] ERROR: /app/server/setup.sh not found or not executable" >&2
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "[bgbye] Starting API..."
|
||
|
|
exec uvicorn server.server:app --host 0.0.0.0 --port 7001 --no-server-header
|
||
|
|
|