diff --git a/docker-compose.yaml b/docker-compose.yaml
index 7a00974..42ac3d6 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -89,16 +89,13 @@ services:
     networks:
       - shared_net
 
-  chromadb:
-    container_name: chromadb
-    image: ${CHROMADB:-chromadb/chroma:latest}
-    restart: always
+  chroma-ui:
+    container_name: chroma-ui
+    image: ghcr.io/logspace-ai/chroma-ui:latest
     ports:
-      - 8000:8000
-    volumes:
-      - chromadb:/chroma  # Persist database storage
+      - "5173:5173"
     environment:
-      IS_PERSISTENT: TRUE  # Enable persistent storage
+      - CHROMA_API_URL=http://chromadb:8000
     networks:
       - shared_net