LocalMathOCR/docker-compose.yml at main · xiaoyeTC/LocalMathOCR · 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
services:
  backend-cpu:
    profiles: ["cpu"]
    build:
      context: ./backend
      dockerfile: Dockerfile.cpu
    container_name: local-math-ocr-backend-cpu
    environment:
      APP_DEVICE: cpu
      DATABASE_URL: sqlite+aiosqlite:///./data/history.db
      MODEL_DIR: /app/models
      CORS_ORIGINS: http://localhost:5173,http://localhost:8080
      DEFAULT_MODEL_ID: pix2tex
      ENABLE_PIX2TEX: "true"
      ENABLE_LATEX_OCR: "true"
      ENABLE_UNI_EQUATION: "false"
      MAX_LOADED_MODELS: "1"
      PRELOAD_MODELS: pix2tex
    volumes:
      - backend-data:/app/data
      - model-cache:/app/models
    ports:
      - "8000:8000"

  backend-gpu:
    profiles: ["gpu"]
    build:
      context: ./backend
      dockerfile: Dockerfile.gpu
    container_name: local-math-ocr-backend-gpu
    environment:
      APP_DEVICE: cuda
      DATABASE_URL: sqlite+aiosqlite:///./data/history.db
      MODEL_DIR: /app/models
      CORS_ORIGINS: http://localhost:5173,http://localhost:8080
      DEFAULT_MODEL_ID: pix2tex
      ENABLE_PIX2TEX: "true"
      ENABLE_LATEX_OCR: "true"
      ENABLE_UNI_EQUATION: "false"
      MAX_LOADED_MODELS: "1"
      PRELOAD_MODELS: pix2tex
    volumes:
      - backend-data:/app/data
      - model-cache:/app/models
    ports:
      - "8000:8000"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

  frontend:
    profiles: ["cpu", "gpu"]
    build:
      context: ./frontend
      dockerfile: Dockerfile
      args:
        VITE_API_BASE_URL: /api
    container_name: local-math-ocr-frontend
    ports:
      - "8080:80"
    extra_hosts:
      - "host.docker.internal:host-gateway"

volumes:
  backend-data:
  model-cache: