Update Dockerfile

Dockerfile

@@ -3,6 +3,9 @@ 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:8b
+
 # Create a new user and switch to it
 RUN useradd -m -u 1000 user
 USER user
@@ -22,4 +25,4 @@ COPY --chown=user . /app
 EXPOSE 7860
 
 # Start Ollama and FastAPI together
-CMD ["sh", "-c", "ollama serve & ollama pull llama3:8b && uvicorn app:app --host 0.0.0.0 --port 7860"]
+CMD ["sh", "-c", "uvicorn app:app --host 0.0.0.0 --port 7860"]