2023-01-05 21:03:53 +00:00
|
|
|
FROM docker.io/python:3.10-buster
|
|
|
|
|
|
|
|
WORKDIR /onnx-web
|
|
|
|
|
2023-01-06 20:17:06 +00:00
|
|
|
# TODO: set up venv within the container
|
|
|
|
|
|
|
|
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
2023-01-05 21:03:53 +00:00
|
|
|
COPY requirements.txt /onnx-web/requirements.txt
|
|
|
|
|
|
|
|
RUN pip install -r requirements.txt
|
|
|
|
|
2023-01-06 22:11:41 +00:00
|
|
|
COPY onnx_web/ /onnx-web/onnx_web/
|
2023-01-05 21:03:53 +00:00
|
|
|
|
2023-01-06 22:11:41 +00:00
|
|
|
ENTRYPOINT [ "flask", "--app=onnx_web", "run", "--host=0.0.0.0" ]
|