apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(build): run convert script when container starts

This commit is contained in:
Sean Sube 2023-01-21 15:10:59 -06:00
parent 45a3ddc2a9
commit a8769a5919
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/Containerfile.cpu.buster
View File

@ -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" ]

2
api/Containerfile.cuda.ubuntu
View File

@ -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" ]

2
api/Containerfile.directml.buster
View File

@ -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" ]

2
api/Containerfile.rocm.ubuntu
View File

@ -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" ]