diff --git a/services/our/.dockerignore b/services/our/.dockerignore
index fd26c8f..01f06dc 100644
--- a/services/our/.dockerignore
+++ b/services/our/.dockerignore
@@ -1,6 +1,5 @@
-!dist/vibeui.pt
-
-runs
+dist
+vibeui
 venv
 src/test
 
diff --git a/services/our/.gitignore b/services/our/.gitignore
index 19c3c96..aefb0a8 100644
--- a/services/our/.gitignore
+++ b/services/our/.gitignore
@@ -1,4 +1,3 @@
-runs
 venv
 vibeui
 
diff --git a/services/our/Dockerfile b/services/our/Dockerfile
index 6c791ae..0101f14 100644
--- a/services/our/Dockerfile
+++ b/services/our/Dockerfile
@@ -21,7 +21,8 @@ COPY --from=ipfs/kubo:v0.36.0 /usr/local/bin/ipfs /usr/local/bin/ipfs
 
 # Bundle the vibeui pytorch model
 RUN mkdir -p /app/dist/vibeui \
-  && wget -q https://gitea.futureporn.net/futureporn/fp/raw/branch/main/apps/vibeui/best.pt -O /app/dist/vibeui/vibeui.pt
+  && wget -q https://gitea.futureporn.net/futureporn/fp/raw/branch/main/apps/vibeui/dist/vibeui.pt -O /app/vibeui/vibeui.pt \
+  && wget -q https://gitea.futureporn.net/futureporn/fp/raw/branch/main/apps/vibeui/dist/data.yaml -O /app/vibeui/data.yaml
 
 # Copy and install dependencies
 COPY package.json package-lock.json ./
diff --git a/services/our/src/utils/vibeui.ts b/services/our/src/utils/vibeui.ts
index a75b2d2..c275fbb 100644
--- a/services/our/src/utils/vibeui.ts
+++ b/services/our/src/utils/vibeui.ts
@@ -139,7 +139,7 @@ export async function inference(videoFilePath: string): Promise<string> {
 
   // Generate a unique name based on video name + UUID
   const uniqueName = nanoid()
-  const customProjectDir = 'runs' // or any custom folder
+  const customProjectDir = 'vibeui/runs' // or any custom folder
   const outputPath = join(env.APP_DIR, customProjectDir, uniqueName)
 
   await spawn('./venv/bin/yolo', [