diff --git a/leafweb-vision/docker-compose.yml b/leafweb-vision/docker-compose.yml new file mode 100644 index 0000000..a3b00ac --- /dev/null +++ b/leafweb-vision/docker-compose.yml @@ -0,0 +1,29 @@ +services: + leafweb-vision: + image: ultralytics/ultralytics:latest + container_name: leafweb-vision + restart: unless-stopped + command: yolo streamlit-predict + ports: + - "8501:8501" + volumes: + - /srv/leafweb-vision/data:/data + - /srv/leafweb-vision/runs:/root/runs + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: all + capabilities: [gpu] + environment: + - TZ=America/New_York + - STREAMLIT_SERVER_ADDRESS=0.0.0.0 + - STREAMLIT_SERVER_PORT=8501 + - STREAMLIT_SERVER_HEADLESS=true + networks: + - npm-network + +networks: + npm-network: + external: true