From cade2b18663ce7d356a4977ea4337c2597dfe0a8 Mon Sep 17 00:00:00 2001
From: Dylan De Faoite <dylanseandefaoite@gmail.com>
Date: Tue, 3 Mar 2026 15:19:57 +0000
Subject: [PATCH] build: fix directories in docker compose

---
 docker-compose.dev.yml |  2 +-
 docker-compose.yml     | 20 +++++++++++++++++---
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/docker-compose.dev.yml b/docker-compose.dev.yml
index 04d435b..ea5e341 100644
--- a/docker-compose.dev.yml
+++ b/docker-compose.dev.yml
@@ -8,7 +8,7 @@ services:
     ports:
       - "5432:5432"
     volumes:
-      - ./server/db/postgres_vol:/var/lib/postgresql/data
+      - ${POSTGRES_DIR}:/var/lib/postgresql/data
       - ./server/db/schema.sql:/docker-entrypoint-initdb.d/schema.sql
 
   redis:
diff --git a/docker-compose.yml b/docker-compose.yml
index baf8048..290b9eb 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -8,7 +8,7 @@ services:
     ports:
       - "5432:5432"
     volumes:
-      - ./server/db/postgres_vol:/var/lib/postgresql/data
+      - ${POSTGRES_DIR}:/var/lib/postgresql/data
       - ./server/db/schema.sql:/docker-entrypoint-initdb.d/schema.sql
 
   redis:
@@ -21,6 +21,8 @@ services:
   backend:
     build: .
     container_name: crosspost_flask
+    volumes:
+      - model_cache:/models
     env_file:
       - .env
     ports:
@@ -32,13 +34,25 @@ services:
 
   worker:
     build: .
+    volumes:
+      - model_cache:/models
     container_name: crosspost_worker
     env_file:
       - .env
     command: >
       celery -A server.queue.celery_app.celery worker
-      --loglevel=info
+      --loglevel=warning
       --pool=solo
     depends_on:
       - postgres
-      - redis
\ No newline at end of file
+      - redis
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
+
+volumes:
+  model_cache:
\ No newline at end of file