feat(build): run convert script when container starts

ssube · Jan 21, 2023 · a8769a5 · a8769a5
1 parent 45a3ddc
commit a8769a5
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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" ]