dream

1
name: toxic-classification-ru
2
endpoints:
3
- respond
4
- respond_batch
5
compose:
6
  env_file:
7
  - .env_ru
8
  build:
9
    context: ./annotators/toxic_classification_ru/
10
    args:
11
      SERVICE_PORT: 8118
12
      PRETRAINED_MODEL_NAME_OR_PATH: s-nlp/russian_toxicity_classifier
13
      LANGUAGE: RU
14
      CUDA_VISIBLE_DEVICES: '0'
15
      FLASK_APP: server
16
  command: flask run -h 0.0.0.0 -p 8118
17
  environment:
18
  - CUDA_VISIBLE_DEVICES=0
19
  - FLASK_APP=server
20
  deploy:
21
    resources:
22
      limits:
23
        memory: 3G
24
      reservations:
25
        memory: 3G
26
  volumes:
27
  - ./annotators/toxic_classification_ru:/src
28
  - ~/.deeppavlov/cache:/root/.cache
29
  ports:
30
  - 8118:8118
31
proxy: null
32