diff --git a/docker-compose.yml b/docker-compose.yml
index f291117..4adae02 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -2,12 +2,12 @@ version: '3.6'
 
 services:
   llama-gpt-api:
-    image: 'ghcr.io/getumbrel/llama-gpt-api-llama-2-7b-chat:latest'
+    image: ghcr.io/getumbrel/llama-gpt-api
     environment:
       MODEL: '/models/llama-2-7b-chat.bin'
 
   llama-gpt-ui:
-    image: 'ghcr.io/getumbrel/llama-gpt-ui:latest'
+    image: ghcr.io/getumbrel/llama-gpt-ui
     ports:
       - 3000:3000
     environment: