diff --git a/ai/compose.yaml b/ai/compose.yaml
index 0a19c4e..0dcd7f1 100644
--- a/ai/compose.yaml
+++ b/ai/compose.yaml
@@ -160,10 +160,12 @@ services:
   facefusion:
     image: ${AI_FACEFUSION_IMAGE:-facefusion/facefusion:3.5.0-cpu}
     container_name: ${AI_COMPOSE_PROJECT_NAME}_facefusion
+    restart: unless-stopped
     entrypoint: ["/entrypoint.sh"]
-    command: ['python', '-u', 'facefusion.py', 'run']
     environment:
       TZ: ${TIMEZONE:-Europe/Berlin}
+      GRADIO_SERVER_NAME: "0.0.0.0"
+      GRADIO_SERVER_PORT: "7860"
     volumes:
       - ai_facefusion_data:/workspace
       - ./entrypoint.sh:/entrypoint.sh:ro
diff --git a/ai/entrypoint.sh b/ai/entrypoint.sh
index 814b938..a2cd939 100755
--- a/ai/entrypoint.sh
+++ b/ai/entrypoint.sh
@@ -13,4 +13,4 @@ else
 fi
 
 echo "Starting Facefusion..."
-cd /facefusion && exec python -u facefusion.py run --listen 0.0.0.0 --port 7860
+cd /facefusion && exec python -u facefusion.py run