Update Dockerfile

Dockerfile

@@ -6,14 +6,14 @@ RUN apt update && apt install -y curl git python3 python3-pip
 # Install Ollama
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Pull the specified Llama 3.2 model
-RUN ollama pull llama3.2:3b-instruct-fp16
+# Start Ollama server in the background and pull the model
+RUN ollama serve & sleep 5 && ollama pull llama3.2:3b-instruct-fp16
 
 # Install Python dependencies
-RUN pip install langchain langchain_ollama
+RUN pip install langchain langchain_ollama ollama
 
 # Expose Ollama API port
 EXPOSE 11434
 
-# Run Ollama on start
+# Start Ollama on container startup
 CMD ["ollama", "serve"]