From d6a5602e1c656a9b3ed06abe5eb5f1b4865881c8 Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Thu, 24 Nov 2022 17:55:47 -0300 Subject: [PATCH] fix docker --- Dockerfile | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/Dockerfile b/Dockerfile index 86e1252..29b94e0 100644 --- a/Dockerfile +++ b/Dockerfile @@ -10,7 +10,7 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machi RUN apt update -y RUN apt upgrade -y -RUN apt install -y curl software-properties-common +RUN apt install -y curl wget software-properties-common RUN add-apt-repository ppa:deadsnakes/ppa RUN apt install -y python3.9 python3.9-distutils RUN curl https://bootstrap.pypa.io/get-pip.py | python3.9 @@ -21,10 +21,10 @@ COPY . . RUN python3.9 -m pip install .[gpu] RUN mkdir -p ~/.u2net -RUN gdown https://drive.google.com/uc?id=1tNuFmLv0TSNDjYIkjEdeH1IWKQdUA4HR -O ~/.u2net/u2netp.onnx -RUN gdown https://drive.google.com/uc?id=1tCU5MM1LhRgGou5OpmpjBQbSrYIUoYab -O ~/.u2net/u2net.onnx -RUN gdown https://drive.google.com/uc?id=1ZfqwVxu-1XWC1xU1GHIP-FM_Knd_AX5j -O ~/.u2net/u2net_human_seg.onnx -RUN gdown https://drive.google.com/uc?id=15rKbQSXQzrKCQurUjZFg8HqzZad8bcyz -O ~/.u2net/u2net_cloth_seg.onnx +RUN wget https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2netp.onnx -O ~/.u2net/u2netp.onnx +RUN wget https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx -O ~/.u2net/u2net.onnx +RUN wget https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx -O ~/.u2net/u2net_human_seg.onnx +RUN wget https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_cloth_seg.onnx -O ~/.u2net/u2net_cloth_seg.onnx EXPOSE 5000 ENTRYPOINT ["rembg"]