Updated dockerfiles to include models in the right directory

akai-katto · Dec 12, 2022 · c96094f · c96094f
1 parent fde0a07
commit c96094f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -77,6 +77,7 @@ RUN pip3.8 install -r /dandere2x/dandere2x/src/requirements.txt
 
 RUN apt -y install libopencv-dev libopencv-imgcodecs-dev libopencv-imgproc-dev libopencv-core-dev
 COPY --from=BASEWAIFUX2CPP /dandere2x/waifu2x-converter-cpp/out /dandere2x/dandere2x/src/externals/waifu2x-converter-cpp
+COPY --from=BASEWAIFUX2CPP /dandere2x/waifu2x-converter-cpp/models_rgb /usr/local/share/waifu2x-converter-cpp/
 
 WORKDIR /dandere2x/dandere2x/src/
 ENTRYPOINT ["python3.8", "/dandere2x/dandere2x/src/main.py"]

diff --git a/Dockerfile2 b/Dockerfile2
@@ -71,6 +71,7 @@ RUN pip3 install -r /dandere2x/dandere2x/src/requirements.txt
 
 RUN apt -y install libopencv-dev libopencv-imgcodecs-dev libopencv-imgproc-dev libopencv-core-dev
 COPY --from=BASEWAIFUX2CPP /dandere2x/waifu2x-converter-cpp/out /dandere2x/dandere2x/src/externals/waifu2x-converter-cpp
+COPY --from=BASEWAIFUX2CPP /dandere2x/waifu2x-converter-cpp/models_rgb /usr/local/share/waifu2x-converter-cpp/
 
 WORKDIR /dandere2x/dandere2x/src/
 ENTRYPOINT ["python3", "/dandere2x/dandere2x/src/main.py"]