diff --git a/leafweb-vision/docker-compose.yml b/leafweb-vision/docker-compose.yml
index a3b00ac..c90499a 100644
--- a/leafweb-vision/docker-compose.yml
+++ b/leafweb-vision/docker-compose.yml
@@ -3,12 +3,13 @@ services:
     image: ultralytics/ultralytics:latest
     container_name: leafweb-vision
     restart: unless-stopped
-    command: yolo streamlit-predict
+    command: bash -c "pip install 'streamlit>=1.29.0' -q --root-user-action=ignore && yolo solutions inference"
     ports:
       - "8501:8501"
     volumes:
       - /srv/leafweb-vision/data:/data
       - /srv/leafweb-vision/runs:/root/runs
+      - pip_cache:/root/.cache/pip
     deploy:
       resources:
         reservations:
@@ -27,3 +28,6 @@ services:
 networks:
   npm-network:
     external: true
+
+volumes:
+  pip_cache: