54 lines
1.5 KiB
YAML
54 lines
1.5 KiB
YAML
version: "3.8"
|
|
|
|
services:
|
|
doc-processer:
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
container_name: doc-processer
|
|
ports:
|
|
- "8053:8053"
|
|
environment:
|
|
- PADDLEOCR_VL_URL=http://host.docker.internal:8080/v1
|
|
- DOCLAYOUT_MODEL_PATH=/app/models/DocLayout/doclayout_yolo_docstructbench_imgsz1024.pt
|
|
- PP_DOCLAYOUT_MODEL_DIR=/app/models/PP-DocLayout
|
|
- MAX_IMAGE_SIZE_MB=10
|
|
volumes:
|
|
# Mount pre-downloaded models (adjust paths as needed)
|
|
- ./models/DocLayout:/app/models/DocLayout:ro
|
|
- ./models/PP-DocLayout:/app/models/PP-DocLayout:ro
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu]
|
|
restart: unless-stopped
|
|
healthcheck:
|
|
test: ["CMD", "curl", "-f", "http://localhost:8053/health"]
|
|
interval: 30s
|
|
timeout: 10s
|
|
retries: 3
|
|
start_period: 10s
|
|
|
|
# Optional: Local development without GPU
|
|
doc-processer-cpu:
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
container_name: doc-processer-cpu
|
|
ports:
|
|
- "8054:8053"
|
|
environment:
|
|
- PADDLEOCR_VL_URL=http://host.docker.internal:8080/v1
|
|
- DOCLAYOUT_MODEL_PATH=/app/models/DocLayout/doclayout_yolo_docstructbench_imgsz1024.pt
|
|
- PP_DOCLAYOUT_MODEL_DIR=/app/models/PP-DocLayout
|
|
volumes:
|
|
- ./models/DocLayout:/app/models/DocLayout:ro
|
|
- ./models/PP-DocLayout:/app/models/PP-DocLayout:ro
|
|
profiles:
|
|
- cpu
|
|
restart: unless-stopped
|
|
|