mateoluksenberg commited on
Commit
1bbdb1c
1 Parent(s): 6474987

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +11 -8
Dockerfile CHANGED
@@ -1,11 +1,14 @@
1
- FROM python:3.9
 
2
 
3
- WORKDIR /code
 
 
 
4
 
5
- COPY ./requirements.txt /code/requirements.txt
 
 
6
 
7
- RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
-
9
- COPY . .
10
-
11
- CMD ["panel", "serve", "/code/app.py", "--address", "0.0.0.0", "--port", "7860", "--allow-websocket-origin", "sophiamyang-panel-example.hf.space", "--allow-websocket-origin", "0.0.0.0:7860"]
 
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"]