fix: dockerfile update

Dockerfile

@@ -11,16 +11,21 @@ RUN pip install --no-cache-dir "earth2studio==0.5.0"
 RUN pip install --no-cache-dir cartopy mlflow
 
 
-RUN useradd -m -u 1001 user
+USER root
+# Create non-root user and set up directories
+RUN useradd -m -u 1001 user && \
+    mkdir -p /home/user/.cache /home/user/.config /home/user/.local /home/user/.local/share/jupyter && \
+    chmod -R 777 /home/user && \
+    mkdir /mhd-demo && chown user:user /mhd-demo && chmod 777 /mhd-demo
+
 USER user
 ENV HOME=/home/user
-RUN mkdir $HOME/.cache $HOME/.config $HOME/.local && chmod -R 777 $HOME
 ENV PATH=/home/user/.local/bin:$PATH
 
-USER user
 # Install required python packages
 RUN pip3 install gdown ipympl cdsapi
 RUN pip3 install --upgrade nbconvert
+
 WORKDIR /e2s
 
 COPY --chown=user Earth2Studio_P1.ipynb /e2s/