Spaces:
Sleeping
Sleeping
mateoluksenberg
commited on
Commit
•
1bbdb1c
1
Parent(s):
6474987
Update Dockerfile
Browse files- Dockerfile +11 -8
Dockerfile
CHANGED
@@ -1,11 +1,14 @@
|
|
1 |
-
|
|
|
2 |
|
3 |
-
|
|
|
|
|
|
|
4 |
|
5 |
-
|
|
|
|
|
6 |
|
7 |
-
|
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"]
|
|
|
|
|
|