From a8769a5919ead33d3328ff75ca09bc68168933ab Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 21 Jan 2023 15:10:59 -0600 Subject: [PATCH] feat(build): run convert script when container starts --- api/Containerfile.cpu.buster | 2 +- api/Containerfile.cuda.ubuntu | 2 +- api/Containerfile.directml.buster | 2 +- api/Containerfile.rocm.ubuntu | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) 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" ]