Update ollama.sh

ollama.sh

@@ -1,30 +1,19 @@
 #!/bin/bash
 
-# Function to start Ollama server
-start_server() {
-    echo "Starting Ollama server"
-    ollama serve &
-    SERVER_PID=$!
-}
+# Start Ollama server in background
+echo "🚀 Starting Ollama server..."
+ollama serve &
 
-# Function to wait for server readiness
-wait_for_server() {
-    echo "Waiting for server to start..."
-    while ! nc -z localhost 11434; do
-        sleep 1
-    done
-}
+# Wait for Ollama to be ready
+echo "⏳ Waiting for Ollama to start..."
+while ! nc -z localhost 11434; do
+    sleep 1
+done
 
-# Function to pull model
-pull_model() {
-    echo "Pulling all-minilm model"
-    ollama pull all-minilm
-}
+# Optional: Pull default model
+echo "📥 Pulling default model..."
+ollama pull all-minilm
 
-# Main execution
-start_server
-wait_for_server
-pull_model
-
-# Keep container running by waiting for server process
-wait $SERVER_PID
+# Start FastAPI reverse proxy
+echo "🔀 Starting FastAPI reverse proxy on port 7860..."
+exec python -m uvicorn ollama:app --host 0.0.0.0 --port 7860 --proxy-headers