diff --git a/docker-compose.yml b/docker-compose.yml index b5a7d61..3721180 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -75,8 +75,13 @@ services: # The live fine-tuned model behind the Classifier bot. GPU-only — opt in with # `--profile gpu`. Uses the psyc-trainer image (built from Dockerfile.train). + # The build context is local so `docker compose --profile gpu build inference` + # actually builds it (without this, compose silently skips the build). inference: image: psyc-trainer + build: + context: . + dockerfile: Dockerfile.train command: ["/scripts/serve_model.py", "--adapter", "/data/adapters/psyc-v5/final", "--host", "0.0.0.0", "--port", "8771"] volumes: - ./data:/data