Update Dockerfile

Dockerfile

@@ -15,6 +15,7 @@ RUN apt-get update && apt-get install -y \
     python3-venv \
     wget \
     ca-certificates \
+    sudo \
     && rm -rf /var/lib/apt/lists/*
 
 # Create and activate virtual environment
@@ -29,10 +30,9 @@ RUN pip install --no-cache-dir --upgrade pip && \
 # Install Ollama
 RUN curl -fsSL https://ollama.ai/install.sh | sh
 
-# Create ollama user and directories
-RUN useradd -m -s /bin/bash ollama && \
-    mkdir -p /home/ollama/.ollama && \
-    chown -R ollama:ollama /home/ollama/.ollama
+# Create app directory and set permissions
+RUN mkdir -p /app/.ollama && \
+    chmod 755 /app/.ollama
 
 # Copy application files
 COPY app.py .
@@ -41,9 +41,10 @@ COPY startup.sh .
 # Make startup script executable
 RUN chmod +x startup.sh
 
-# Set environment variables
+# Set environment variables for Ollama
 ENV OLLAMA_HOST=0.0.0.0:11434
-ENV OLLAMA_MODELS=/home/ollama/.ollama/models
+ENV OLLAMA_MODELS=/app/.ollama/models
+ENV OLLAMA_HOME=/app/.ollama
 
 # Expose ports
 EXPOSE 7860 11434