diff --git a/docker-compose.yaml b/docker-compose.yaml
index 7d4131e..8831ee0 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -10,9 +10,14 @@ services:
     env_file:
       - .env
     volumes:
-      - ~/Downloads/sd-v1-4-full-ema.ckpt:/app/model.ckpt
+      - ./images:/images
+      - ./notebooks:/notebooks
+      - ./models:/root/.cache/huggingface/diffusers/
+    ports:
+      - 8888:8888
+      - 8000:8000
 
   vosk:
     image: alphacep/kaldi-en
+    ports:
       - 2700:2700
-    runtime: nvidia
\ No newline at end of file