diff --git a/docker-compose.yaml b/docker-compose.yaml
index 8b62df4..cdb7d2c 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -13,6 +13,7 @@ services:
       - ~/Downloads/sd-v1-4-full-ema.ckpt:/app/model.ckpt
 
   vosk:
-    image: alphacep/kaldi-en
+    image: alphacep/kaldi-en-gpu
     ports:
-      - 2700:2700
\ No newline at end of file
+      - 2700:2700
+    runtime: nvidia
\ No newline at end of file