version: '3.8' | |
services: | |
traffic-detection: | |
build: . | |
ports: | |
- "7860:7860" | |
volumes: | |
- ./ANPR_IND:/app/ANPR_IND | |
- ./Helmet-Detect-model:/app/Helmet-Detect-model | |
- ./logs:/app/logs | |
environment: | |
- PYTHONUNBUFFERED=1 | |
- GRADIO_SERVER_NAME=0.0.0.0 | |
- GRADIO_SERVER_PORT=7860 | |
restart: unless-stopped | |
healthcheck: | |
test: ["CMD", "curl", "-f", "http://localhost:7860/api/health"] | |
interval: 30s | |
timeout: 10s | |
retries: 3 | |
deploy: | |
resources: | |
reservations: | |
devices: | |
- driver: nvidia | |
count: 1 | |
capabilities: [gpu] |