Update entrypoint.sh

entrypoint.sh

@@ -1,20 +1,29 @@
 #!/bin/bash
 
-# Set the OLLAMA_ROOT environment variable to a writable directory
+# Set the OLLAMA_ROOT to a writable directory
 export OLLAMA_ROOT=/tmp/ollama
 
 # Start the Ollama server in the background
+echo "Starting Ollama server..."
 ollama serve &
+pid=$!
 
 # Wait for the server to be ready
-sleep 5
+echo "Waiting for server to start..."
+while ! curl -s -o /dev/null http://127.0.0.1:11434; do
+  echo "Server not yet available, waiting..."
+  sleep 1
+done
+echo "Server is up and running."
 
-# Pull the necessary models
-#ollama pull deepseek-r1:1.5b
+# Pull the models
+echo "Pulling models..."
+ollama pull gemma3:1b
 ollama pull gemma3:270m
-ollama pull granite-embedding
 ollama pull nomic-embed-text
-ollama pull gemma3:1b
 
-# Keep the container running
-tail -f /dev/null
+echo "All models pulled."
+
+# Wait for the server process to exit
+# This brings the background process to the foreground
+wait $pid