diff --git a/ai/entrypoint.sh b/ai/entrypoint.sh
index 8b3a7ec..4326b63 100755
--- a/ai/entrypoint.sh
+++ b/ai/entrypoint.sh
@@ -1,10 +1,13 @@
 #!/bin/sh
-# Start background job to continuously delete NSFW models
-(
-  while true; do
-    rm -f /facefusion/.assets/models/nsfw_*.onnx /facefusion/.assets/models/nsfw_*.hash 2>/dev/null
-    sleep 2
-  done
-) &
-# Run the original command
-exec python -u facefusion.py run --config-path /config/facefusion-config.ini
+# Wait for models to be downloaded, then delete NSFW models once
+python -u facefusion.py run --config-path /config/facefusion-config.ini &
+PID=0
+
+# Wait 60 seconds for initial model downloads
+sleep 60
+
+# Delete NSFW models
+rm -f /facefusion/.assets/models/nsfw_*.onnx /facefusion/.assets/models/nsfw_*.hash
+
+# Wait for the main process
+wait