Update entrypoint.sh

entrypoint.sh

@@ -1,15 +1,25 @@
 #!/bin/bash
 
-echo "Starting Ollama server..."
+# Function to pull and run a model
+pull_and_run_model() {
+    local model_name=$1
+    echo "Pulling $model_name"
+    ollama pull "$model_name"
+    echo "Running $model_name"
+    ollama run "$model_name" --keepalive -1s &
+}
+
+# Starting server
+echo "Starting server"
 ollama serve &
+sleep 5  # Allowing time for the server to start
 
-echo "Waiting for Ollama server to be active..."
-while ! ollama list | grep -q 'NAME'; do
-  sleep 1
+# Splitting the models by comma and processing each
+IFS=',' read -ra MODELS <<< "$model"
+for m in "${MODELS[@]}"; do
+    pull_and_run_model "$m"
+    sleep 5  # Optional: Delay between starting each model
 done
 
-echo "Pulling the specified model..."
-ollama pull "${MODEL_NAME}"
-
-echo "Model pulled successfully. Keeping the server running..."
-wait
+# Keep the script running to prevent the container from exiting
+wait