Update Dockerfile

Dockerfile

@@ -1,18 +1,25 @@
 FROM ollama/ollama:latest
+
+# Stay as root to make changes
 USER root
 
-# Add user `ollama` only if UID 1000 is not taken; fallback to random UID otherwise
-RUN if id -u 1000 >/dev/null 2>&1; then \
-        echo "UID 1000 already exists. Using a random UID..."; \
-        uid=$(( ( RANDOM % 63000 )  + 2000 )); \
-        useradd -m -u $uid ollama && \
-        echo "Created ollama with UID $uid"; \
+# Create ollama user with a consistent UID and ensure proper ownership
+RUN if id -u ollama >/dev/null 2>&1; then \
+        echo "ollama user already exists"; \
     else \
-        useradd -m -u 1000 ollama && \
-        echo "Created ollama with UID 1000"; \
-    fi && \
-    mkdir -p /home/ollama/.ollama && \
-    chown -R ollama:ollama /home/ollama/.ollama
+        if id -u 1000 >/dev/null 2>&1; then \
+            echo "UID 1000 already exists. Using UID 1001..."; \
+            useradd -m -u 1001 -s /bin/bash ollama; \
+        else \
+            useradd -m -u 1000 -s /bin/bash ollama; \
+        fi \
+    fi
+
+# Create necessary directories and set proper ownership
+RUN mkdir -p /home/ollama/.ollama/models && \
+    mkdir -p /home/ollama/.ollama/logs && \
+    chown -R ollama:ollama /home/ollama && \
+    chmod -R 755 /home/ollama
 
 # Set environment variables
 ENV OLLAMA_HOST=0.0.0.0:7860 \
@@ -21,11 +28,15 @@ ENV OLLAMA_HOST=0.0.0.0:7860 \
     OLLAMA_MAX_LOADED_MODELS=1 \
     OLLAMA_CONTEXT_LENGTH=8196 \
     OLLAMA_KEEP_ALIVE=1m \
-    OLLAMA_CPU_THREADS=4
+    OLLAMA_CPU_THREADS=4 \
+    HOME=/home/ollama
 
-# Use the created user
+# Switch to ollama user
+USER ollama
 WORKDIR /home/ollama
 
+# Expose port
 EXPOSE 7860
 
+# Start Ollama service
 CMD ["serve"]