commit2 runpod

docker-compose.yml

@@ -0,0 +1,17 @@
+version: '3.8'
+
+services:
+  vr180-matting:
+    build: .
+    runtime: nvidia
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all
+      - CUDA_VISIBLE_DEVICES=0
+    volumes:
+      - ./data:/app/data          # Mount data directory
+      - ./output:/app/output      # Mount output directory
+      - ./models:/app/models      # Mount models directory
+    working_dir: /app
+    stdin_open: true
+    tty: true
+    command: /bin/bash