diff --git a/docker-compose.yml b/docker-compose.yml
index 4a35b5a5e..2cb7131e8 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -20,8 +20,12 @@ services:
     environment:
       - MODEL_SERVER=http://llmserver:8000
       - LOCAL_DB_HOST=db
+      - WEB_SERVER_PORT=7860
+    volumes:
+      - ./models:/app/models
+      - data:/app/pilot/data
     env_file:
-      - .env
+      - .env.template
     ports:
       - 7860:7860
     expose:
@@ -31,10 +35,12 @@ services:
     build:
       context: .
       dockerfile: Dockerfile-llmserver
+    environment:
+      - LOCAL_DB_HOST=db
     volumes:
       - ./models:/app/models
     env_file:
-      - .env    
+      - .env.template
     ports:
       - 8000:8000
     restart: unless-stopped
@@ -43,10 +49,11 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              device_ids: ['1', '0']
+              device_ids: ['1']
               capabilities: [gpu]
 
 
 
 volumes:
   my-db:
+  data: