t5-summarizer-api/docker-compose.yaml at main · pheepa/t5-summarizer-api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
version: '3'
services:
  app:
    build:
      context: .
      dockerfile: app/Dockerfile
    command: python main.py
    volumes:
      - .:/code
    ports:
      - "80:80"
  celery_worker:
    build:
      context: .
      dockerfile: celery_worker/Dockerfile
    command: celery -A celery_worker.worker worker -l info --concurrency 1 -P solo
    environment:
      - BROKER_URI=amqp://rmuser:rmpassword@rabbitmq:5672//
#      - BACKEND_URI=rpc://
      - BACKEND_URI=db+sqlite:///results.db
      - MODEL_PATH=celery_worker/models/t5-base-jira-pubmed-finetuned
      - MAX_SOURCE_LENGTH=2048
      - MAX_TARGET_LENGTH=256
      - TASK_PREFIX=summarize
      - DEVICE=cpu
    volumes:
      - .:/code
    depends_on:
      - app
      - rabbitmq
      - db
  rabbitmq:
    image: rabbitmq:3.11-management-alpine
    hostname: rabbitmq
    ports:
      - "15672:15672"
      - "5672:5672"
    environment:
      - RABBITMQ_DEFAULT_USER=rmuser
      - RABBITMQ_DEFAULT_PASS=rmpassword
    healthcheck:
      test: [ "CMD", "nc", "-z", "localhost", "5672" ]
      interval: 5s
      timeout: 15s
      retries: 1
  db:
    image: nouchka/sqlite3:latest
    volumes:
      - ./db:/db
#    restart: always
volumes:
  code: