ml-xraylab/docker-compose.yaml at master · Minyewoo/ml-xraylab · 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
69
70
71
72
73
74
version: "3"
services:
  rabbitmq:
    hostname: rabbitmq
    image: rabbitmq:3.9.25-management-alpine
    ports:
      - 5672:5672
      - 15672:15672
    volumes:
      - ./data/rabbitmq/:/var/lib/rabbitmq/
    networks:
      - rabbitmq_net
    healthcheck:
      test: ["CMD", "rabbitmq-diagnostics", "-q", "ping"]
      interval: 30s
      timeout: 30s
      retries: 5

  inference:
    build: ./ml/
    environment:
      - RABBITMQ_HOST=rabbitmq
      - INFERENCE_EXCHANGE_NAME=inference
      - WEIGHTS_PATH=/app/weights/efficientnetb3
      - API_URL=http://app:5000
      - API_USER=inference
      - API_PASSWORD=d38a068fd7943a39c6feef268f065e5132eb9d3328485339f6d81b282d2e235a
    volumes:
      - ./ml/weights/:/app/weights/:ro
    restart: on-failure
    depends_on:
      - rabbitmq
    networks:
      - rabbitmq_net
      - api_net

  app:
    hostname: app
    build: ./api/
    environment:
      - RABBITMQ_HOST=rabbitmq
      - INFERENCE_EXCHANGE_NAME=inference
      - UPLOAD_FOLDER=static
    ports:
      - 4999:5000
    volumes:
      - ./api/:/app/
      # - ./api/static:/app/static
    restart: on-failure
    networks:
      - mysql_net
      - api_net
      - rabbitmq_net

  db:
    image: mysql:8.0.30
    hostname: db
    environment:
      - MYSQL_ROOT_PASSWORD=root
    restart: on-failure
    volumes:
      - ./db:/docker-entrypoint-initdb.d/:ro
    networks:
      - mysql_net

networks:
  rabbitmq_net:
    driver: bridge

  mysql_net:
    driver: bridge

  api_net:
    driver: bridge