mateoluksenberg commited on
Commit
ac06001
1 Parent(s): 39591e8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -7
Dockerfile CHANGED
@@ -1,14 +1,12 @@
1
- # Usa una imagen base de Python con soporte para GPU (si es necesario)
2
- FROM nvidia/cuda:11.0-base
3
 
4
  # Instala las bibliotecas necesarias
5
- RUN apt-get update && \
6
- apt-get install -y python3-pip python3-dev && \
7
- pip3 install transformers pillow
8
 
9
  # Copia tu script de aplicación y el modelo dentro del contenedor
10
  COPY app.py .
11
  COPY cm5.jpg .
12
 
13
- # Comando por defecto para ejecutar tu aplicación
14
- CMD ["python3", "app.py"]
 
1
+ # Usa una imagen base de Python
2
+ FROM python:3.8-slim
3
 
4
  # Instala las bibliotecas necesarias
5
+ RUN pip install transformers pillow
 
 
6
 
7
  # Copia tu script de aplicación y el modelo dentro del contenedor
8
  COPY app.py .
9
  COPY cm5.jpg .
10
 
11
+ # Comando por defecto para ejecutar tu aplicación cuando se inicie el contenedor
12
+ CMD ["python", "app.py"]