dream

1
name: midas-classification
2
endpoints:
3
- model
4
- respond
5
- batch_model
6
compose:
7
  env_file:
8
  - .env
9
  build:
10
    args:
11
      SERVICE_PORT: 8090
12
      SERVICE_NAME: midas_classification
13
      CONFIG: midas_conv_bert.json
14
      CUDA_VISIBLE_DEVICES: '0'
15
      FLASK_APP: server
16
    context: ./annotators/midas_classification
17
  command: flask run -h 0.0.0.0 -p 8090
18
  environment:
19
  - CUDA_VISIBLE_DEVICES=0
20
  - FLASK_APP=server
21
  deploy:
22
    resources:
23
      limits:
24
        memory: 3G
25
      reservations:
26
        memory: 3G
27
  volumes:
28
  - ./annotators/midas_classification:/src
29
  - ~/.deeppavlov:/root/.deeppavlov
30
  ports:
31
  - 8090:8090
32