diff --git a/docker-compose.yml b/docker-compose.yml
index 1012af8..3e7eca9 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -2,6 +2,9 @@ services:
   comfyui:
     build: .
     container_name: ai-comfyui
+    volumes:
+      - ./data/models:/app/comfyui/models
+      - ./data/output:/app/comfyui/output
     ports:
       - 8188:8188
     restart: always
@@ -11,5 +14,4 @@ services:
           devices:
             - driver: nvidia
               count: all
-              capabilities: [gpu]
-
+              capabilities: [gpu]
\ No newline at end of file