From 2f91297ff900d859456179c5fa9c65a46d528d16 Mon Sep 17 00:00:00 2001
From: Jimmy <git@jimmy.nz>
Date: Tue, 1 Nov 2022 13:59:54 +1300
Subject: [PATCH] Expose ports, Add volumes

---
 docker-compose.yaml | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

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