Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
 FROM nvidia/cuda:11.3.1-base-ubuntu20.04
 
 ENV DEBIAN_FRONTEND=noninteractive \
-	TZ=Europe/Paris
+    TZ=Europe/Paris
 
 # Remove any third-party apt sources to avoid issues with expiring keys.
 # Install some basic utilities
@@ -26,7 +26,7 @@ RUN add-apt-repository ppa:flexiondotorg/nvtop && \
     apt-get upgrade -y && \
     apt-get install -y --no-install-recommends nvtop
 
-RUN curl -sL https://deb.nodesource.com/setup_14.x  | bash - && \
+RUN curl -sL https://deb.nodesource.com/setup_14.x | bash - && \
     apt-get install -y nodejs && \
     npm install -g configurable-http-proxy
 
@@ -39,17 +39,22 @@ RUN adduser --disabled-password --gecos '' --shell /bin/bash user \
 RUN echo "user ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/90-user
 USER user
 
-# All users can use /home/user as their home directory
-ENV HOME=/home/user
-RUN mkdir $HOME/.cache $HOME/.config \
+# Change HOME to /data
+ENV HOME=/data
+RUN mkdir -p $HOME/.cache $HOME/.config \
  && chmod -R 777 $HOME
 
+# Ensure /data directory exists and is accessible
+USER root
+RUN mkdir -p /data && chown -R user:user /data
+USER user
+
 # Set up the Conda environment
 ENV CONDA_AUTO_UPDATE_CONDA=false \
     PATH=$HOME/miniconda/bin:$PATH
 RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py39_4.10.3-Linux-x86_64.sh \
  && chmod +x ~/miniconda.sh \
- && ~/miniconda.sh -b -p ~/miniconda \
+ && ~/miniconda.sh -b -p $HOME/miniconda \
  && rm ~/miniconda.sh \
  && conda clean -ya
 
@@ -62,14 +67,14 @@ WORKDIR $HOME/app
 USER root
 
 # User Debian packages
-## Security warning : Potential user code executed as root (build time)
+## Security warning: Potential user code executed as root (build time)
 RUN --mount=target=/root/packages.txt,source=packages.txt \
     apt-get update && \
     xargs -r -a /root/packages.txt apt-get install -y --no-install-recommends \
     && rm -rf /var/lib/apt/lists/*
 
 RUN --mount=target=/root/on_startup.sh,source=on_startup.sh,readwrite \
-	bash /root/on_startup.sh
+    bash /root/on_startup.sh
 
 #######################################
 # End root user section
@@ -89,11 +94,11 @@ RUN chmod +x start_server.sh
 COPY --chown=user login.html /home/user/miniconda/lib/python3.9/site-packages/jupyter_server/templates/login.html
 
 ENV PYTHONUNBUFFERED=1 \
-	GRADIO_ALLOW_FLAGGING=never \
-	GRADIO_NUM_PORTS=1 \
-	GRADIO_SERVER_NAME=0.0.0.0 \
-	GRADIO_THEME=huggingface \
-	SYSTEM=spaces \
-	SHELL=/bin/bash
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
+    SYSTEM=spaces \
+    SHELL=/bin/bash
 
 CMD ["./start_server.sh"]