Update Dockerfile

Dockerfile

@@ -1,23 +1,25 @@
-FROM ubuntu:22.04
+FROM ollama/ollama:latest
 
-ENV DEBIAN_FRONTEND=noninteractive
-ENV OLLAMA_HOST=0.0.0.0:7860
+# Create a user and set up permissions
+RUN useradd -ms /bin/bash ollama-user
+ENV HOME=/home/ollama-user
+WORKDIR $HOME
+
+# Ensure the user has access to the directory
+RUN mkdir -p $HOME/.ollama && chown -R ollama-user:ollama-user $HOME/.ollama
 
-# Install dependencies (including zstd — the missing piece)
-RUN apt-get update && \
-    apt-get install -y curl ca-certificates zstd && \
-    rm -rf /var/lib/apt/lists/*
+# Install netcat (nc) for checking server readiness
+RUN apt-get update && apt-get install -y netcat
 
-# Install Ollama (official script)
-RUN curl -fsSL https://ollama.com/install.sh | sh
+# Copy the entrypoint script
+COPY entrypoint.sh /usr/local/bin/entrypoint.sh
+RUN chmod +x /usr/local/bin/entrypoint.sh
 
+# Switch to the non-root user
+USER ollama-user
+
+# Set Ollama to listen on the correct port
+ENV OLLAMA_HOST=0.0.0.0:7860
 EXPOSE 7860
 
-CMD bash -c "\
-  ollama serve & \
-  until curl -s http://localhost:7860 > /dev/null; do \
-    sleep 1; \
-  done; \
-  ollama pull qwen3.5:0.8b; \
-  wait"
-  
+ENTRYPOINT ["/usr/local/bin/entrypoint.sh"]