diff --git a/docker-compose.yml b/docker-compose.yml
index 3be13ef..1be4649 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,18 +1,17 @@
 services:
   whisper:
+    image: rocm-wyoming-whisper
     container_name: whisper
     restart: unless-stopped
-    build:
-      context: .
     group_add:
       - video
     volumes:
-      - ./data:/data
+      - whisper-data:/data
     command:
       - --download-dir
       - /data
-      - --language
-      - en
+      - --model
+      - medium
       - --device
       - cuda
     ports:
@@ -20,3 +19,6 @@ services:
     devices:
       - "/dev/dri"
       - "/dev/kfd"
+
+volumes:
+  whisper-data:
\ No newline at end of file