Create start.sh

start.sh

@@ -0,0 +1,41 @@
+#!/bin/bash
+set -e
+
+ # Function to log messages with timestamp
+log() {
+    echo "[$(date '+%Y-%m-%d %H:%M:%S')] $1"
+}
+
+ # Function to check if a service is up
+wait_for_service() {
+    local host=$1
+    local port=$2
+    local retries=30
+    local wait=2
+
+     log "Waiting for $host:$port to be available..."
+
+     for ((i=1;i<=retries;i++)); do
+        if nc -z "$host" "$port"; then
+            log "$host:$port is available."
+            return 0
+        else
+            log "Attempt $i/$retries: $host:$port not available. Retrying in $wait seconds..."
+            sleep "$wait"
+        fi
+    done
+
+     log "Error: $host:$port did not become available after $((retries * wait)) seconds."
+    exit 1
+}
+
+ # Start TorchServe
+log "Starting TorchServe..."
+torchserve --start --ncs --ts-config /home/torchserve/model-store
+
+ # Wait until TorchServe is up and running
+wait_for_service "localhost" 8080
+
+ # Start Gradio app
+log "Starting Gradio application..."
+python app.py