Update Dockerfile

Dockerfile

@@ -1,16 +1,24 @@
-# usar imagen base de Python 3.10
-FROM python:3.10
+# Etapa 1: Instalación de dependencias de Python y construcción de la aplicación
+FROM python:3.10 AS builder
 
-# instalar dependencias de Python
-RUN pip install pandas numpy nltk moviepy pydub transformers pydantic pytest whisperx fastapi uvicorn
+# Copiar archivos de la aplicación
+WORKDIR /app
+COPY . .
+
+# Instalar dependencias de Python
+RUN pip install --no-cache-dir pandas numpy nltk moviepy pydub transformers pydantic pytest fastapi uvicorn torch torchvision torchaudio
 
-# instalar paquete especial whisperx desde GitHub
+# Instalar paquete especial whisperx desde GitHub
 RUN pip install git+https://github.com/m-bain/whisperx.git
 
-# instalar paquetes de PyTorch para CUDA 12.1
-RUN pip install torch==1.10.0+cu121 torchvision==0.11.1+cu121 torchaudio==0.10.0+cu121 -f https://download.pytorch.org/whl/torch_stable.html
+# Instalar dependencias desde requirements.txt
+COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+# Etapa 2: Agregar componentes específicos de CUDA y utilidades básicas
+FROM nvidia/cuda:11.3.1-base-ubuntu20.04 AS final
 
-# instalar utilidades básicas
+# Instalar utilidades básicas
 RUN apt-get update && apt-get install -y \
     curl \
     git \
@@ -20,19 +28,12 @@ RUN apt-get update && apt-get install -y \
     git-lfs \
     ffmpeg
 
-# copiar archivos de la aplicación
-COPY . /app
-# copiar todos los archivos de la aplicación
-
-COPY . .
-# establecer directorio de trabajo
+# Copiar la aplicación desde la etapa anterior
 WORKDIR /app
+COPY --from=builder /app .
 
-# instalar dependencias de Python desde requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
-
-# cambiar permisos de archivos necesarios
+# Cambiar permisos de archivos necesarios
 RUN chmod -R 777 /app
 
-# definir comando predeterminado para iniciar la aplicación
+# Definir comando predeterminado para iniciar la aplicación
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]