From 52195378cc4bf0f8a83b0c086cfef088a108447b Mon Sep 17 00:00:00 2001
From: Daniel Ramos <46768340+danielcamposramos@users.noreply.github.com>
Date: Sun, 26 Jan 2025 21:46:53 -0300
Subject: [PATCH] Update Dockerfile

Added cpu torch installation.
---
 transcription-api/Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/transcription-api/Dockerfile b/transcription-api/Dockerfile
index e680f9d..c99cc05 100644
--- a/transcription-api/Dockerfile
+++ b/transcription-api/Dockerfile
@@ -11,6 +11,7 @@ RUN export DEBIAN_FRONTEND=noninteractive \
 COPY requirements.txt requirements.txt
 RUN pip3 install -r requirements.txt
 RUN pip3 install python-multipart
+RUN pip3 install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
 
 COPY . .
 
@@ -18,4 +19,4 @@ HEALTHCHECK --interval=30s --timeout=10s --start-period=15s \
 CMD ["python3", "healthcheck.py"]
 
 EXPOSE 8000
-CMD ["python3", "main.py"]
\ No newline at end of file
+CMD ["python3", "main.py"]