dream

1
name: entity-detection
2
endpoints:
3
- respond
4
- respond_batch
5
compose:
6
  env_file:
7
  - .env
8
  build:
9
    args:
10
      SERVICE_NAME: entity_detection
11
      SEQ_TAG_CONFIG: wikipedia_entity_detection_distilbert.json
12
      CONFIG: entity_detection_eng.json
13
      LOWERCASE: 1
14
      SERVICE_PORT: 8103
15
      SRC_DIR: annotators/entity_detection/
16
      FINEGRAINED: 0
17
      FLASK_APP: server
18
      CUDA_VISIBLE_DEVICES: '0'
19
    context: ./
20
    dockerfile: annotators/entity_detection/Dockerfile
21
  command: flask run -h 0.0.0.0 -p 8103
22
  environment:
23
  - FLASK_APP=server
24
  - CUDA_VISIBLE_DEVICES=0
25
  deploy:
26
    resources:
27
      limits:
28
        memory: 2.5G
29
      reservations:
30
        memory: 2.5G
31
  volumes:
32
  - ./annotators/entity_detection:/src
33
  - ~/.deeppavlov:/root/.deeppavlov
34
  - ~/.deeppavlov/cache:/root/.cache
35
  ports:
36
  - 8103:8103
37
proxy:
38
  command:
39
  - nginx
40
  - -g
41
  - daemon off;
42
  build:
43
    context: dp/proxy/
44
    dockerfile: Dockerfile
45
  environment:
46
  - PROXY_PASS=dream.deeppavlov.ai:8103
47
  - PORT=8103
48