17 lines
398 B
Plaintext
17 lines
398 B
Plaintext
FROM docker.io/python:3.10-alpine
|
|
|
|
RUN apk add --update cmake \
|
|
&& rm -rf /var/cache/apk/*
|
|
|
|
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
|
|
|
|
RUN pip install -r requirements.txt
|
|
|
|
COPY serve.py /onnx-web/serve.py
|
|
|
|
ENTRYPOINT [ "flask", "--app=serve", "run", "--host=0.0.0.0" ]
|