Allow stopping of inference

run.sh

@@ -2,6 +2,15 @@
 
 set -e
 
+function chown_iterations() {
+    if [ -d "./iterations" ] && [ "$(find ./iterations/ ! -user $USER -o ! -group $USER 2>/dev/null)" ]; then
+        echo "Chowning iterations directory"
+        sudo chown -R $USER:$USER ./iterations/
+    fi
+}
+
+trap chown_iterations EXIT
+
 docker build \
     --tag sia \
     .
@@ -13,6 +22,6 @@ docker run \
     -p 8080:8080 \
     -v /$(pwd)/model/:/root/model/ \
     -v /$(pwd)/.env:/root/.env \
-    -v /$(pwd)/iterations/:/root/sia/iterations/ \
+    -v /$(pwd)/iterations/:/root/iterations/ \
     -v /$(pwd)/.git/:/root/sia/.git/ \
-    sia "$@"
+    sia "$@"