# Usa una imagen base de Python con soporte para GPU (si es necesario)
FROM nvidia/cuda:11.0-base

# Instala las bibliotecas necesarias
RUN apt-get update && \
    apt-get install -y python3-pip python3-dev && \
    pip3 install transformers pillow

# Copia tu script de aplicación y el modelo dentro del contenedor
COPY app.py .
COPY cm5.jpg .

# Comando por defecto para ejecutar tu aplicación
CMD ["python3", "app.py"]