diff --git a/docker-compose.yml b/docker-compose.yml
index 31166bc..a2417cf 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -33,6 +33,8 @@ services:
       - "5001:5001"
       - "8080:8080"
     # network_mode: host
+    environment:
+      - YOLO_MODELS="${YOLO_MODELS}"
 
 volumes:
   frigate-config: