1
0
Fork 0

fix(api): improve image layer order

This commit is contained in:
Sean Sube 2023-01-06 14:17:06 -06:00
parent af40bff10c
commit 54829786bd
2 changed files with 10 additions and 6 deletions

View File

@ -5,11 +5,12 @@ RUN apk add --update cmake \
WORKDIR /onnx-web
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
COPY requirements.txt /onnx-web/requirements.txt
COPY serve.py /onnx-web/serve.py
RUN pip install -r requirements.txt
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
COPY serve.py /onnx-web/serve.py
ENTRYPOINT [ "flask", "--app=serve", "run", "--host=0.0.0.0" ]

View File

@ -2,11 +2,14 @@ FROM docker.io/python:3.10-buster
WORKDIR /onnx-web
COPY requirements.txt /onnx-web/requirements.txt
COPY serve.py /onnx-web/serve.py
RUN pip install -r requirements.txt
# TODO: set up venv within the container
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
COPY requirements.txt /onnx-web/requirements.txt
RUN pip install -r requirements.txt
COPY serve.py /onnx-web/serve.py
ENTRYPOINT [ "flask", "--app=serve", "run", "--host=0.0.0.0" ]