diff --git a/api/Containerfile.cpu.buster b/api/Containerfile.cpu.buster index b38dbbe9..c19cedb6 100644 --- a/api/Containerfile.cpu.buster +++ b/api/Containerfile.cpu.buster @@ -28,4 +28,4 @@ ENV ONNX_WEB_BUNDLE_PATH="/onnx-web/gui" COPY gui/ /onnx-web/gui/ -CMD [ "flask", "--app=onnx_web.serve", "run", "--host=0.0.0.0" ] +CMD [ "sh", "-c", "python -m onnx_web.convert --diffusion --upscaling --correction --extras && flask --app=onnx_web.serve run --host=0.0.0.0" ] diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu index c182d108..04713332 100644 --- a/api/Containerfile.cuda.ubuntu +++ b/api/Containerfile.cuda.ubuntu @@ -26,4 +26,4 @@ ENV ONNX_WEB_BUNDLE_PATH="/onnx-web/gui" COPY gui/ /onnx-web/gui/ -CMD [ "flask", "--app=onnx_web.serve", "run", "--host=0.0.0.0" ] +CMD [ "sh", "-c", "python -m onnx_web.convert --diffusion --upscaling --correction --extras && flask --app=onnx_web.serve run --host=0.0.0.0" ] diff --git a/api/Containerfile.directml.buster b/api/Containerfile.directml.buster index 433653e6..f57c16ca 100644 --- a/api/Containerfile.directml.buster +++ b/api/Containerfile.directml.buster @@ -26,4 +26,4 @@ ENV ONNX_WEB_BUNDLE_PATH="/onnx-web/gui" COPY gui/ /onnx-web/gui/ -CMD [ "flask", "--app=onnx_web.serve", "run", "--host=0.0.0.0" ] +CMD [ "sh", "-c", "python -m onnx_web.convert --diffusion --upscaling --correction --extras && flask --app=onnx_web.serve run --host=0.0.0.0" ] diff --git a/api/Containerfile.rocm.ubuntu b/api/Containerfile.rocm.ubuntu index 9c13dc56..9adf9e8b 100644 --- a/api/Containerfile.rocm.ubuntu +++ b/api/Containerfile.rocm.ubuntu @@ -20,4 +20,4 @@ ENV ONNX_WEB_BUNDLE_PATH="/onnx-web/gui" COPY gui/ /onnx-web/gui/ -CMD [ "flask", "--app=onnx_web.serve", "run", "--host=0.0.0.0" ] +CMD [ "sh", "-c", "python -m onnx_web.convert --diffusion --upscaling --correction --extras && flask --app=onnx_web.serve run --host=0.0.0.0" ]