fridge-vision / docker-compose.yml
NoNameWrath's picture
Deploy to HuggingFace Spaces
c02c6ce
Raw
History Blame Contribute Delete
579 Bytes
version: '3.8'
services:
fridge-vision-api:
build: .
container_name: fridge-vision
ports:
- "8000:8000"
environment:
- MODEL_NAME=yolov5s
- CONF_THRESHOLD=0.5
- API_HOST=0.0.0.0
- API_PORT=8000
- LOG_LEVEL=INFO
- DEBUG=false
volumes:
- ./data:/app/data
- ./logs:/app/logs
restart: unless-stopped
# Uncomment for GPU support
# deploy:
# resources:
# reservations:
# devices:
# - driver: nvidia
# count: 1
# capabilities: [gpu]