一键部署:主程序+OCR同机(screen/GPU),Ollama外置局域网。
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
#!/usr/bin/env bash
|
||||
# 供 screen 会话调用:OCR 模型常驻 GPU,异常退出自动重启
|
||||
set -uo pipefail
|
||||
|
||||
ROOT="$(cd "$(dirname "$0")" && pwd)"
|
||||
VENV="${ROOT}/.venv"
|
||||
PORT="${OCR_PORT:-23567}"
|
||||
|
||||
export OCR_USE_GPU="${OCR_USE_GPU:-true}"
|
||||
export OCR_HOST="${OCR_HOST:-0.0.0.0}"
|
||||
|
||||
if [[ ! -d "${VENV}" ]]; then
|
||||
echo "错误: 请先运行 bash install.sh"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# shellcheck disable=SC1091
|
||||
source "${VENV}/bin/activate"
|
||||
cd "${ROOT}"
|
||||
|
||||
echo "=========================================="
|
||||
echo " OCR Worker | GPU=${OCR_USE_GPU} | :${PORT}"
|
||||
echo " $(date -Iseconds)"
|
||||
echo " 退出 screen: Ctrl+A 然后 D"
|
||||
echo "=========================================="
|
||||
|
||||
while true; do
|
||||
uvicorn app:app --host "${OCR_HOST}" --port "${PORT}" --log-level info
|
||||
code=$?
|
||||
echo "[$(date -Iseconds)] uvicorn 退出 code=${code},5 秒后重启…"
|
||||
sleep 5
|
||||
done
|
||||
Reference in New Issue
Block a user