updated docker

Dockerfile

@@ -13,7 +13,7 @@ RUN apt-get update && apt-get install -y \
 # Install Ollama
 RUN curl -fsSL https://ollama.ai/install.sh | sh
 
-# Copy requirements and install
+# Copy requirements and install dependencies
 COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
@@ -37,8 +37,8 @@ COPY --chown=user . $HOME/app
 # Ensure Ollama has a cache directory
 RUN mkdir -p /tmp/ollama_cache
 
-# Pull the correct LLaMA model
-RUN ollama pull llama3.2
+# Start Ollama in the background and pull the model
+RUN ollama serve > /tmp/ollama.log 2>&1 & sleep 5 && ollama pull llama3.2
 
 # Start Ollama before running FastAPI
 CMD ollama serve > /tmp/ollama.log 2>&1 & sleep 5 && uvicorn main:app --host 0.0.0.0 --port 7860