Update Dockerfile

Dockerfile

@@ -4,19 +4,16 @@ FROM python:3.8-slim
 # Set the working directory inside the container
 WORKDIR /app
 
-# Install system dependencies and curl for Ollama installation
+# Install system dependencies, curl, and any other dependencies
 RUN apt-get update && apt-get install -y \
     curl \
     && pip install --upgrade pip
 
-# Install Ollama using the provided script
+# Install Ollama
 RUN curl -sSL https://ollama.com/install.sh | bash
 
-# Pull the Llama 3.2 model from Ollama
-RUN ollama pull llama-3.2
-
 # Expose the port for the model server (assuming default port 5000)
 EXPOSE 5000
 
-# Command to start the Ollama server with the llama-3.2 model
-CMD ["ollama", "serve", "--model", "llama-3.2", "--port", "5000"]
+# Pull the Llama 3.2 model - this will happen when the container starts
+CMD /bin/bash -c "ollama pull llama-3.2 && ollama serve --model llama-3.2 --port 5000"