adding signal handler

Author: rjtokenring

containers/ei-models-runner/src/start-runner.sh

@@ -1,19 +1,17 @@
 #!/bin/bash
 
-echo "$@"
-
 NODE_COMMAND=("node" "/app/linux/node/build/cli/linux/runner.js" "$@")
 
-#trap_signal() {
-#  echo "Caught signal $1. Exiting wrapper immediately..."
-#  exit 0
-#}
+trap_signal() {
+  echo "Caught signal $1. Exiting wrapper immediately..."
+  exit 0
+}
 
-#trap 'trap_signal TERM' TERM
-#trap 'trap_signal INT' INT
+trap 'trap_signal TERM' TERM
+trap 'trap_signal INT' INT
 
 while true; do
-  echo "🚀 Starting EI runner..."
+  echo "🚀 Starting EI inference runner..."
 
   "${NODE_COMMAND[@]}"