mlbusters/docker-compose.yml at master · RedmanPlus/mlbusters · 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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
version: "3.10"

services:
  db:
    image: chromadb/chroma:latest
    volumes:
      - chroma-data:/chroma/chroma
    restart: unless-stopped
    ports:
      - "8000:8000"

  cache:
    image: memcached:latest
    ports:
      - "11211:11211"
    restart: always

  encode:
    build:
      context: ./inference
      dockerfile: Dockerfile
    command: uvicorn main:app --host "0.0.0.0" --port 8040 --log-config=log_conf.yaml
    restart: unless-stopped
    volumes:
      - inference-model-data:/app/model_data
    env_file:
      - inference/.env.dist
    ports:
      - "8040:8040"

  search:
    build:
      context: ./inference
      dockerfile: Dockerfile
    command: uvicorn main:app --host "0.0.0.0" --port 8050 --log-config=log_conf.yaml
    restart: unless-stopped
    volumes:
      - inference-model-data:/app/model_data
    env_file:
      - inference/.env.dist
    ports:
      - "8050:8050"

  main:
    build:
      context: ./main
      dockerfile: Dockerfile
    command: uvicorn main:app --host "0.0.0.0" --port 80
    restart: unless-stopped
    volumes:
      - main-model-data:/app/model_data
    depends_on:
      - db
      - encode
      - search
      - cache
    env_file:
      - main/.env.dist
    ports:
      - "80:80"

volumes:
  chroma-data:
    driver: local
  inference-model-data:
    driver: local
  main-model-data:
    driver: local