Update startup.sh

startup.sh

@@ -1,19 +1,20 @@
 #!/bin/bash
+set -e
 
-# Start Ollama server in the background
+# 1. Launch Ollama server in background
 ollama serve &
 OLLAMA_PID=$!
+trap "kill $OLLAMA_PID" EXIT
 
-# Wait for Ollama server to become available
+# 2. Wait until the API port is open
 echo "⏳ Waiting for Ollama to become ready..."
-
-for i in {1..20}; do
+for i in {1..30}; do
   if curl -s http://localhost:11434 > /dev/null; then
     echo "✅ Ollama is up."
     break
   else
     echo "⏳ Attempt $i: Ollama not ready yet..."
-    sleep 1
+    sleep 2
   fi
 done
 
@@ -22,13 +23,10 @@ if ! curl -s http://localhost:11434 > /dev/null; then
   exit 1
 fi
 
-# Pull the model
-echo "📦 Pulling model: unsloth/gemma-3n-E2B-it-GGUF..."
-ollama pull gemma:2b
+# 3. Pull a real model
+echo "📦 Pulling model: gemma:2b"
+ollama pull gemma2:2b || { echo "❌ Failed to pull model"; exit 1; }
 
-# Start your Gradio app
+# 4. Launch Gradio app
 echo "🚀 Launching app..."
-python main.py
-
-# Wait for Ollama to exit (optional for cleanup)
-wait $OLLAMA_PID
+python main.py