.

Dockerfile

@@ -18,20 +18,8 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
-# Switch to the "user" user
-USER user
-# Set home to the user's home directory
-ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH
-
-# Set the working directory to the user's home directory
-WORKDIR $HOME/app
-
-# Copy the current directory contents into the container at $HOME/app setting the owner to the user
-COPY --chown=user . $HOME/app
 
 # Create nginx configuration for WebSocket support using echo commands
-USER root
 RUN echo 'server {' > /etc/nginx/sites-available/default && \
     echo '    listen 7860;' >> /etc/nginx/sites-available/default && \
     echo '    server_name _;' >> /etc/nginx/sites-available/default && \
@@ -62,9 +50,7 @@ RUN echo 'server {' > /etc/nginx/sites-available/default && \
 # Give user sudo permissions for nginx
 RUN echo "user ALL=(ALL) NOPASSWD: /usr/sbin/nginx, /usr/sbin/nginx -t, /usr/sbin/nginx -s quit" >> /etc/sudoers
 
-USER user
-
-# Create a startup script for HF Spaces with nginx
+# Create a startup script that runs as root initially (while we're still root)
 RUN echo '#!/bin/bash' > /start_hf_spaces.sh && \
     echo 'set -e' >> /start_hf_spaces.sh && \
     echo '' >> /start_hf_spaces.sh && \
@@ -162,9 +148,17 @@ RUN echo '#!/bin/bash' > /start_hf_spaces.sh && \
     echo 'kill $TAIL_PID 2>/dev/null || true' >> /start_hf_spaces.sh && \
     chmod +x /start_hf_spaces.sh
 
-# Remove the old user-level startup script
+# Switch to user and copy files
 USER user
-RUN rm -f start_hf_spaces.sh
+# Set home to the user's home directory
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
 
 # Switch back to root for final startup
 USER root