Update Dockerfile

Dockerfile

@@ -17,9 +17,7 @@ RUN pip install -r requirements.txt
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
 # Start Ollama server in the background and pull the required model
-RUN ollama serve & sleep 5
-RUN ollama pull llama3.2:latest
-RUN ollama pull llama3.2:latest
+CMD ["python", "ollama_serve.py", "llama3.2"]
 
 # Set the command to run the Chainlit application
 CMD ["chainlit", "run", "app.py", "--port", "7860"]