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: