Subida de código

whisper-distributed/processor/Dockerfile

@@ -0,0 +1,29 @@
+FROM python:3.9
+
+WORKDIR /app
+
+# Instalar dependencias básicas
+RUN apt-get update && apt-get install -y ffmpeg
+
+# Instalar torch con soporte CUDA y otras dependencias
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+RUN pip install pika openai-whisper
+
+# Crear un script para descargar el modelo
+RUN echo '\
+import whisper\n\
+import torch\n\
+print("Verificando disponibilidad de CUDA:", torch.cuda.is_available())\n\
+if torch.cuda.is_available():\n\
+    print("Dispositivo CUDA:", torch.cuda.get_device_name(0))\n\
+print("Descargando modelo Whisper base...")\n\
+whisper.load_model("base")\n\
+print("Modelo descargado correctamente")\
+' > download_model.py
+
+# Ejecutar el script para descargar el modelo durante la construcción
+RUN python download_model.py
+
+COPY app.py .
+
+CMD ["python", "app.py"]