Improved the system prompt

test.sh

@@ -1,26 +1,12 @@
 #!/bin/bash
 
-# Continue on error
-set -e
-
-# Build with progress output and capture the tag
-TAG=$( \
 docker build \
     --target sia-test \
-    . \
-    2>&1 | tee /dev/tty | grep "writing image" | cut -d' ' -f4 \
-)
+    --tag sia-test \
+    .
 
-# Exit if tag is empty
-[ -z "$TAG" ] && exit 1
-
-# Run tests
 docker run \
     --rm \
     --gpus=all \
-    --privileged \
     -v /$(pwd)/model/:/root/model/ \
-    $TAG
-
-# Clean up image
-[ ! -z "$TAG" ] && docker rmi $TAG
+    sia-test