dekun
33 lines
855 B
Bash
33 lines
855 B
Bash
#!/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:-false}"
|
||
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
|