Update Dockerfile

Dockerfile

@@ -1,13 +1,29 @@
+# Use an official lightweight Ubuntu base image
 FROM ubuntu:22.04
 
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive
+ENV OLLAMA_HOME=/home/user/.ollama
+
+# Create a non-root user to avoid permission issues
+RUN useradd -m -s /bin/bash user
+
 # Install dependencies
-RUN apt update && apt install -y curl git
+RUN apt update && apt install -y \
+    curl git && \
+    rm -rf /var/lib/apt/lists/*
 
 # Install Ollama
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Expose API port
+# Ensure the Ollama home directory is writable
+RUN mkdir -p $OLLAMA_HOME && chown -R user:user $OLLAMA_HOME
+
+# Switch to non-root user
+USER user
+
+# Expose Ollama API port
 EXPOSE 11434
 
-# Run Ollama on start
+# Start Ollama server
 CMD ["ollama", "serve"]