Clean images

Softcatala · Jan 5, 2025 · 40d48be · 40d48be
1 parent 73fa04f
commit 40d48be
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,11 +1,12 @@
+# Alternatively, use a base image with CUDA and cuDNN support
 FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04
 
 RUN apt-get update && apt-get install -y python3-pip
 
-# Install dependencies
-RUN pip3 install --no-cache-dir -r requirements.txt
-RUN pip install -U whisper-ctranslate2
+WORKDIR /app
+COPY . /app
 
+RUN pip3 install --no-cache-dir -r requirements.txt
+RUN pip install -U .
 
-# Set the entry point
 ENTRYPOINT ["whisper-ctranslate2"]