From e7d979d43ed99e2513c9f8c73bc63f532f8aa9d4 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Thu, 30 May 2024 18:54:09 -0700 Subject: [PATCH] Update Dockerfile --- Dockerfile | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index 7f3d3c0c..660a257f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -25,13 +25,13 @@ WORKDIR /app # Install Python dependencies RUN pip3 install uv && \ if [ "$USE_CUDA" = "true" ]; then \ - # If you use CUDA the whisper and embedding model will be downloaded on first use pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/$USE_CUDA_DOCKER_VER --no-cache-dir && \ - uv pip install --system -r requirements.txt --no-cache-dir \ + uv pip install --system -r requirements.txt --no-cache-dir; \ else \ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu --no-cache-dir && \ - uv pip install --system -r requirements.txt --no-cache-dir \ - fi; + uv pip install --system -r requirements.txt --no-cache-dir; \ + fi + # Expose the port ENV HOST="0.0.0.0"