Upload 4 files

Dockerfile

@@ -1,8 +1,13 @@
 FROM python:3.11-slim
 
+# Create a non-root user
+RUN useradd -m -u 1000 ollama && \
+    mkdir -p /home/ollama/.ollama && \
+    chown -R ollama:ollama /home/ollama
+
 WORKDIR /app
 
-# Install system dependencies including Ollama
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     curl \
     wget \
@@ -18,8 +23,19 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy application code
 COPY . .
 
+# Set proper ownership
+RUN chown -R ollama:ollama /app
+
+# Switch to ollama user
+USER ollama
+
 # Create a startup script
-RUN echo '#!/bin/bash\nollama serve &\nsleep 5\npython3 -m gunicorn --bind 0.0.0.0:7860 --workers 1 --timeout 120 app:app' > /app/start.sh && \
+RUN echo '#!/bin/bash\n\
+export OLLAMA_MODELS=/home/ollama/.ollama\n\
+export OLLAMA_HOST=0.0.0.0\n\
+ollama serve &\n\
+sleep 15\n\
+python3 -m gunicorn --bind 0.0.0.0:7860 --workers 1 --timeout 120 app:app' > /app/start.sh && \
     chmod +x /app/start.sh
 
 # Expose port