Update Dockerfile

Dockerfile

@@ -1,8 +1,13 @@
 # Use the official Python image from the Docker Hub
 FROM python:3.9
 
-# Install curl
-RUN apt-get update && apt-get install -y curl
+# Install curl and Ollama as root
+RUN apt-get update && apt-get install -y curl \
+    && curl -fsSL https://ollama.com/install.sh | sh \
+    && rm -rf /var/lib/apt/lists/*
+
+# Pull the necessary models using Ollama
+RUN ollama pull mistral && ollama pull nomic-embed-text
 
 # Create a new user with a home directory and set the user ID
 RUN useradd -m -u 1000 user
@@ -23,10 +28,5 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 # Copy the rest of the application code to the /app directory
 COPY --chown=user . /app
 
-# Install Ollama
-RUN curl -fsSL https://ollama.com/install.sh | sh
-
-RUN ollama pull mistral
-RUN ollama pull nomic-embed-text
 # Set the command to run the application
 CMD ["uvicorn", "server:app", "--host", "0.0.0.0", "--port", "7860"]