braille-ocr/docker-compose.yml at main · endernoke/braille-ocr · 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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
services:
  redis:
    image: redis:7-alpine@sha256:02f2cc4882f8bf87c79a220ac958f58c700bdec0dfb9b9ea61b62fb0e8f1bfcf
    container_name: redis
    profiles: [app, training]
    ports:
      - "6379:6379"
    volumes:
      - redis_data:/data
    healthcheck:
      test: ["CMD", "redis-cli", "ping"]
      interval: 10s
      timeout: 5s
      retries: 5
    networks:
      - network

  api:
    build:
      context: ./backend
      dockerfile: Dockerfile
    container_name: api
    profiles: [app]
    ports:
      - "8000:8000"
    env_file:
      - .env
    volumes:
      - ./backend/src:/app/src
      - uploads:/app/uploads
      - results:/app/results
    depends_on:
      redis:
        condition: service_healthy
    networks:
      - network
    restart: unless-stopped

  worker:
    build:
      context: ./backend
      dockerfile: Dockerfile.worker
      args:
        - OCR_MODEL_PATH=${OCR_MODEL_PATH}
        - CLASSIFIER_MODEL_PATH=${CLASSIFIER_MODEL_PATH}
    container_name: worker
    profiles: [app]
    env_file:
      - .env
    volumes:
      - ./backend/src:/app/src
      - uploads:/app/uploads
      - results:/app/results
    depends_on:
      redis:
        condition: service_healthy
    networks:
      - network
    restart: unless-stopped

    # deploy:
    #   resources:
    #     reservations:
    #       devices:
    #         - driver: nvidia
    #           count: 1
    #           capabilities: [gpu]

  frontend:
    build:
      context: ./frontend
      dockerfile: Dockerfile
    container_name: frontend
    profiles: [app]
    ports:
      - "3000:80"
    depends_on:
      - api
    networks:
      - network
    restart: unless-stopped

  training:
    build:
      context: .
      dockerfile: ./training/Dockerfile
    container_name: training
    profiles: [training]
    environment:
      - DEVICE=cuda
    volumes:
      - ./training:/app/training
      - ./backend/src:/app/backend/src
      - ./data:/data
      - models:/app/models
    depends_on:
      - redis
    networks:
      - network
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

networks:
  network:
    driver: bridge

volumes:
  redis_data:
  uploads:
  results:
  models: