Update Dockerfile

Dockerfile

@@ -4,6 +4,9 @@ FROM ubuntu:22.04
 # Set the working directory in the container
 WORKDIR /app
 
+# Ensure we are running as root
+USER root
+
 # Install system dependencies and Python
 RUN apt-get update && apt-get install -y \
     python3 \
@@ -36,5 +39,5 @@ COPY . .
 # Expose the FastAPI default port
 EXPOSE 8000
 
-# Start Ollama, pull Llama3 if not present, then start FastAPI
-CMD ["sh", "-c", "ollama serve & sleep 5 && ollama pull llama3 && uvicorn main:app --host 0.0.0.0 --port 8000"]
+# Start Ollama and FastAPI as root
+CMD ["sh", "-c", "export OLLAMA_HOME=/app/.ollama && ollama serve & sleep 5 && ollama pull llama3 && uvicorn main:app --host 0.0.0.0 --port 8000"]