diff --git a/Dockerfile b/Dockerfile index 76781e2..65b69d8 100644 --- a/Dockerfile +++ b/Dockerfile @@ -20,11 +20,8 @@ RUN --mount=type=cache,target=/root/.cache \ --mount=type=ssh,required=true \ pip install --no-cache-dir . -# dependencies for running piper on gpu -RUN --mount=type=cache,target=/root/.cache pip install --no-cache-dir onnxruntime-gpu - # Expose the API port EXPOSE 7999 # Run the translator server -CMD ["python", "-m", "auracast_translator.translator_server.translator_server"] +CMD ["python", "-m", "auracast_translator.translator_server"] diff --git a/docker-compose.yml b/docker-compose.yml index 9f011e6..6811373 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -11,7 +11,7 @@ services: - PYTHONUNBUFFERED=1 restart: unless-stopped - deploy: + deploy: # for tts on gpu resources: reservations: devices: