Update Dockerfile

Dockerfile

@@ -4,22 +4,35 @@ ARG BASE_CPU_CONTAINER=ubuntu:${UBUNTU_VERSION}
 FROM ${BASE_CPU_CONTAINER} as build
 
 RUN apt-get update && \
-    apt-get install -y build-essential git make cmake wget && \
-    git clone https://github.com/ggerganov/llama.cpp.git /build/llama.cpp && \
-    cd /build/llama.cpp && \
-    make llama-server && \
-    wget -nv https://huggingface.co/Green-Sky/TriLM_3.9B-GGUF/resolve/main/TriLM_3.9B_Unpacked-4.0B-TQ2_0.gguf -O /data/model.gguf
+    apt-get install -y build-essential git make cmake wget
+
+WORKDIR /build
+
+RUN git clone https://github.com/ggerganov/llama.cpp.git
+
+WORKDIR /build/llama.cpp
+
+
+ENV LDFLAGS="-static"
+
+RUN make llama-server
+
+WORKDIR /data
+RUN wget https://huggingface.co/Green-Sky/TriLM_3.9B-GGUF/resolve/main/TriLM_3.9B_Unpacked-4.0B-TQ2_0.gguf -nv -O model.gguf
 
 FROM ${BASE_CPU_CONTAINER} as runtime
 
 WORKDIR /app
 
-COPY --from=build /build/llama.cpp/llama-server /app/llama-server
+# Copy the executable from the build stage
+COPY --from=build /build/llama.cpp/llama-server /app
 COPY --from=build /data/model.gguf /data/model.gguf
 COPY ./run.sh /app/run.sh
-
+WORKDIR /app
 EXPOSE 7860
 
+# Make the script executable
 RUN chmod +x run.sh
 
-CMD ["./run.sh"]
+# CMD to run your script
+CMD ./run.sh