diff --git a/Dockerfile b/Dockerfile index 0182438..1f9f56f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,8 +1,7 @@ FROM python:3.9 # download this https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx -# copy model to avoid unnecessary download -COPY u2net.onnx /home/.u2net/u2net.onnx +ADD https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx /root/.u2net/u2net.onnx WORKDIR /app @@ -14,4 +13,4 @@ COPY . . EXPOSE 5100 -CMD ["python", "app.py"] \ No newline at end of file +CMD ["python", "app.py"]