Update Dockerfile

Dockerfile

@@ -1,12 +1,8 @@
-# Use official Python base image
 FROM python:3.12.5
 
 # Install Ollama as root
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Start Ollama in the background and pull the model
-RUN ollama serve & sleep 5 && ollama pull llama3
-
 # Create a new user and switch to it
 RUN useradd -m -u 1000 user
 USER user
@@ -26,4 +22,4 @@ COPY --chown=user . /app
 EXPOSE 7860
 
 # Start Ollama and FastAPI together
-CMD ["sh", "-c", "ollama serve & uvicorn app:app --host 0.0.0.0 --port 7860"]
+CMD ["sh", "-c", "ollama serve & ollama pull llama3:8b && uvicorn app:app --host 0.0.0.0 --port 7860"]