version: '3.8'

services:
  app:
    build: .
    volumes:
      - .:/work
      - ./Data:/Data
      - ./model_assets:/model_assets
    ports:
      - "7878:7860"
    tty: true
    command: python app.py --share --server-name 0.0.0.0

    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [ gpu ]

  webui-train:
    build:
      context: .
      dockerfile: Dockerfile.webui.train
    volumes:
      - .:/work
      - ./Data:/Data
      - ./model_assets:/model_assets
    ports:
      - "8590:8501"
    tty: true
    # command: python app.py --share --server-name 0.0.0.0

    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [ gpu ]