update: iopaint事前ダウンロード

Author: thr3a

iopaint/Dockerfile

@@ -6,5 +6,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 RUN pip install iopaint rembg
 RUN iopaint download --model=lama
-ENTRYPOINT ["iopaint", "start", "--model=lama", "--device=cuda", "--host=0.0.0.0", "--port=8080", "--enable-remove-bg", "--enable-anime-seg"]
+RUN python -c "from iopaint.plugins.anime_seg import AnimeSeg; AnimeSeg()"
 
+ENTRYPOINT ["iopaint"]
+CMD ["start", "--device=cuda", "--host=0.0.0.0", "--port=8080", "--enable-remove-bg", "--remove-bg-model", "briaai/RMBG-1.4"]