fix compose

This commit is contained in:
Maurizio Dipierro 2025-03-19 16:23:45 +01:00
parent 671f475e11
commit be8aaff73c
1 changed files with 4 additions and 3 deletions

View File

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