SmolDocling-256M-Demo/docker-compose.yml

24 lines
611 B
YAML

docling_app:
build:
context: .
args:
ENABLE_CUDA: '1'
container_name: gradio_app
restart: unless-stopped
ports:
- "7880:7860" # Gradio application port
volumes:
- ./app/services/weaviate_service.py:/gradio_app/weaviate_service.py
environment:
ENABLE_CUDA: '1'
NVIDIA_VISIBLE_DEVICES: '3'
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ['3'] # Specify GPU 3
capabilities: [gpu]
shm_size: 1g
depends_on:
- rag_app