diff --git a/machine-learning/start.sh b/machine-learning/start.sh index 36c2b86259..0836213e6a 100755 --- a/machine-learning/start.sh +++ b/machine-learning/start.sh @@ -5,9 +5,11 @@ export LD_PRELOAD="/usr/lib/$(arch)-linux-gnu/libmimalloc.so.2" : "${MACHINE_LEARNING_HOST:=0.0.0.0}" : "${MACHINE_LEARNING_PORT:=3003}" : "${MACHINE_LEARNING_WORKERS:=1}" +: "${MACHINE_LEARNING_WORKER_TIMEOUT:=120}" gunicorn app.main:app \ -k uvicorn.workers.UvicornWorker \ -w $MACHINE_LEARNING_WORKERS \ -b $MACHINE_LEARNING_HOST:$MACHINE_LEARNING_PORT \ + -t $MACHINE_LEARNING_WORKER_TIMEOUT \ --log-config-json log_conf.json