diff --git a/apps/vibeui/public/data.yaml b/apps/vibeui/public/data.yaml
new file mode 100644
index 0000000..a9b48ca
--- /dev/null
+++ b/apps/vibeui/public/data.yaml
@@ -0,0 +1,25 @@
+path: /home/cj/Documents/futureporn-monorepo/apps/vibeui/data
+train: train/images
+val: val/images
+
+names:
+  0: ControlledByTipper
+  1: ControlledByTipperHigh
+  2: ControlledByTipperLow
+  3: ControlledByTipperMedium
+  4: ControlledByTipperUltrahigh
+  5: Earthquake
+  6: Fireworks
+  7: HighLevel
+  8: LowLevel
+  9: MediumLevel
+  10: Pause
+  11: Pulse
+  12: RandomTime
+  13: RespondingTo
+  14: Ring1
+  15: Ring2
+  16: Ring3
+  17: Ring4
+  18: UltraHighLevel
+  19: Wave
diff --git a/apps/vibeui/public/vibeui.pt b/apps/vibeui/public/vibeui.pt
new file mode 100644
index 0000000..9edc264
Binary files /dev/null and b/apps/vibeui/public/vibeui.pt differ
diff --git a/services/our/Dockerfile b/services/our/Dockerfile
index 0101f14..81eed24 100644
--- a/services/our/Dockerfile
+++ b/services/our/Dockerfile
@@ -20,9 +20,9 @@ RUN apt-get update -y && \
 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/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
+RUN mkdir -p /app/vibeui \
+  && wget -q https://gitea.futureporn.net/futureporn/fp/raw/branch/main/apps/vibeui/public/vibeui.pt -O /app/vibeui/vibeui.pt \
+  && wget -q https://gitea.futureporn.net/futureporn/fp/raw/branch/main/apps/vibeui/public/data.yaml -O /app/vibeui/data.yaml
 
 # Copy and install dependencies
 COPY package.json package-lock.json ./