From a65bb40fa7918c06f73a2606243dce88ad799052 Mon Sep 17 00:00:00 2001
From: Robert Dunmire III <slackr31337@gmail.com>
Date: Sun, 19 Jan 2025 20:14:53 -0500
Subject: [PATCH] Add docker-compose.yml

---
 Dockerfile         |  3 ++-
 docker-compose.yml | 22 ++++++++++++++++++++++
 run.sh             |  3 ---
 3 files changed, 24 insertions(+), 4 deletions(-)
 create mode 100644 docker-compose.yml

diff --git a/Dockerfile b/Dockerfile
index 914d1a3..e5a56cc 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -89,7 +89,8 @@ RUN \
 FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 AS dist
 
 ENV PYTHONUNBUFFERED=1
-ENV PATH="/bin:$PATH"
+ENV LD_LIBRARY_PATH="/app/lib:/usr/local/cuda-11.8/targets/x86_64-linux/lib:${LD_LIBRARY_PATH}"
+ENV PATH="/app/bin:${PATH}" 
 
 RUN \
     mkdir -p /data /app &&\
diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..3dc69a6
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,22 @@
+services:
+  wyoming-piper:  
+    image: slackr31337/wyoming-piper-gpu:latest  
+    container_name: wyoming-piper
+    environment:  
+      - PIPER_VOICE="en_US-lessac-medium"
+    ports:
+      - 10300:10300
+    volumes:  
+      - /path/to/persistent/data:/data  
+    restart: unless-stopped
+    runtime: nvidia
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities:
+                - gpu
+                - utility
+                - compute
\ No newline at end of file
diff --git a/run.sh b/run.sh
index 2425436..79fda55 100644
--- a/run.sh
+++ b/run.sh
@@ -1,9 +1,6 @@
 #!/usr/bin/env bash
 
-export LD_LIBRARY_PATH=/app/lib/:/usr/local/cuda-11.8/targets/x86_64-linux/lib:$LD_LIBRARY_PATH
-
 # Run wyoming-piper server
-source /app/bin/activate
 /app/bin/python3 -m wyoming_piper \
     --piper '/app/piper/piper' \
     --uri 'tcp://0.0.0.0:10200' \