diff --git a/docker-compose.yaml b/docker-compose.yaml
index 319c86c..8b62df4 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -5,6 +5,14 @@ services:
     image: imageserver
     build: imageserver
     runtime: nvidia
-    command: python3 /main.py
+    ports:
+      - 8000:8000
     env_file:
-      - .env
\ No newline at end of file
+      - .env
+    volumes:
+      - ~/Downloads/sd-v1-4-full-ema.ckpt:/app/model.ckpt
+
+  vosk:
+    image: alphacep/kaldi-en
+    ports:
+      - 2700:2700
\ No newline at end of file