version: '3.7'

services:
  cornac:
    image: ghcr.io/preferredai/cornac-server:latest
    volumes:
      - $PWD/save_dir:/app/cornac/serving/save_dir
      - cornacvol:/app/cornac/serving/data
    environment:
      - MODEL_PATH=save_dir/bpr
      - MODEL_CLASS=cornac.models.BPR
    ports:
      - 8080:5000
volumes:
  cornacvol: