auto msg

Dockerfile

@@ -68,6 +68,7 @@ EXPOSE 7860
 # Set the environment variables
 ENV JUPYTERLAB_PORT=8888
 ENV NGINX_PORT=7860
+ENV JUPYTER_TOKEN=""
 
 # Run the entrypoint script when the container starts
-ENTRYPOINT ["/app/entrypoint.sh"]
+ENTRYPOINT ["/app/entrypoint.sh"]

entrypoint.sh

@@ -20,12 +20,11 @@ fi
 # Asegurar que el usuario jovyan sea el propietario de /data
 chown -R ${NB_UID}:${NB_GID} /data
 
-# Leer el token de JupyterLab del secreto de HF
-JUPYTERLAB_TOKEN=$(cat /run/secrets/JUPYTER_TOKEN)
-
-# Verificar si el token está vacío
-if [ -z "$JUPYTERLAB_TOKEN" ]; then
-    handle_error "El token de JupyterLab está vacío"
+# Verificar si el secreto JUPYTER_TOKEN existe y asignarlo a la variable de entorno
+if [ -f "/run/secrets/JUPYTER_TOKEN" ]; then
+    export JUPYTER_TOKEN=$(cat /run/secrets/JUPYTER_TOKEN)
+elif [ -z "$JUPYTER_TOKEN" ]; then
+    handle_error "El token de JupyterLab está vacío. Asegúrate de configurar el secreto JUPYTER_TOKEN en la configuración del Space."
 fi
 
 # Verificar disponibilidad de GPU
@@ -39,7 +38,7 @@ fi
 
 # Iniciar JupyterLab en segundo plano
 jupyter lab --ip=0.0.0.0 --port=${JUPYTERLAB_PORT} --no-browser --allow-root \
-    --NotebookApp.base_url=/jupyter --NotebookApp.token=${JUPYTERLAB_TOKEN} \
+    --NotebookApp.base_url=/jupyter --NotebookApp.token=${JUPYTER_TOKEN} \
     --notebook-dir=/data &
 
 # Asegurar que el archivo PID de Nginx sea escribible
@@ -47,4 +46,4 @@ touch /tmp/nginx.pid
 chmod 644 /tmp/nginx.pid
 
 # Iniciar Nginx en primer plano
-nginx -g "daemon off;" -c /etc/nginx/nginx.conf || handle_error "Fallo al iniciar Nginx"
+nginx -g "daemon off;" -c /etc/nginx/nginx.conf || handle_error "Fallo al iniciar Nginx"