Update Dockerfile

Dockerfile

@@ -7,7 +7,10 @@ WORKDIR /app
 RUN apt-get update && apt-get install -y --no-install-recommends     git     cmake     build-essential     libopencv-dev     libgomp1     wget     ca-certificates     ocl-icd-opencl-dev     opencl-headers     pocl-opencl-icd  && rm -rf /var/lib/apt/lists/*
 
 # Build waifu2x-converter-cpp from source
-RUN git clone --depth 1 https://github.com/DeadSix27/waifu2x-converter-cpp.git /tmp/waifu2x     && cd /tmp/waifu2x     && mkdir build && cd build     && cmake -DCMAKE_BUILD_TYPE=Release ..     && make -j$(nproc)     && cp waifu2x-converter-cpp /app/     && mkdir -p /app/models     && cp -r ../models_rgb/* /app/models/     && rm -rf /tmp/waifu2x     && apt-get purge -y git cmake build-essential     && apt-get autoremove -y     && apt-get clean
+RUN git clone --depth 1 https://github.com/DeadSix27/waifu2x-converter-cpp.git /tmp/waifu2x     && cd /tmp/waifu2x     && mkdir build && cd build     && cmake -DCMAKE_BUILD_TYPE=Release ..     && make -j$(nproc)     && cp waifu2x-converter-cpp /app/     && cp libw2xc.so /app/     && mkdir -p /app/models     && cp -r ../models_rgb/* /app/models/     && rm -rf /tmp/waifu2x     && apt-get purge -y git cmake build-essential     && apt-get autoremove -y     && apt-get clean
+
+# Set library path so the executable can find libw2xc.so
+ENV LD_LIBRARY_PATH=/app:$LD_LIBRARY_PATH
 
 # Verify the executable works
 RUN chmod +x /app/waifu2x-converter-cpp