From be8aaff73c302d2e11046666ba21af4294ada013 Mon Sep 17 00:00:00 2001 From: Maurizio Dipierro Date: Wed, 19 Mar 2025 16:23:45 +0100 Subject: [PATCH] fix compose --- docker-compose.yml | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index 8b322fb..6981581 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -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: