Update Dockerfile

Dockerfile

@@ -1,6 +1,9 @@
 # Use official Python base image
 FROM python:3.12.5
 
+# Install Ollama as root
+RUN curl -fsSL https://ollama.com/install.sh | sh
+
 # Create a new user and switch to it
 RUN useradd -m -u 1000 user
 USER user
@@ -13,11 +16,8 @@ WORKDIR /app
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-# Install Ollama
-RUN curl -fsSL https://ollama.com/install.sh | sh
-
 # Pull the Llama3 model (or any other model you need)
-RUN ollama pull llama3:8b
+RUN ollama pull llama3
 
 # Copy application code
 COPY --chown=user . /app