Update Dockerfile

Dockerfile

@@ -1,14 +1,16 @@
-# Use the official lightweight Python image
 FROM python:3.10-slim
 
-# Set environment variables to avoid prompts during installs
+# Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 
-# Set working directory to home/streamlit_user (non-root user directory)
+# Create a non-root user and home directory
+RUN useradd -m -s /bin/bash streamlit_user
+
+# Set working directory
 WORKDIR /home/streamlit_user/app
 
-# Install system dependencies (including for image processing and huggingface)
+# Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     git \
@@ -16,28 +18,26 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libgl1-mesa-glx \
     && rm -rf /var/lib/apt/lists/*
 
-# Create a directory for model weights under the user directory and set permissions
+# Create model weights directory and assign ownership
 RUN mkdir -p /home/streamlit_user/step1x_weights && \
-    chown -R streamlit_user:streamlit_user /home/streamlit_user/step1x_weights
+    chown -R streamlit_user:streamlit_user /home/streamlit_user
 
-# Copy requirements.txt first to leverage Docker cache
+# Copy requirements and install Python dependencies
 COPY requirements.txt ./requirements.txt
-
-# Install Python dependencies
 RUN pip install --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of your app files
+# Copy app source code
 COPY src/ ./src/
 
-# Set user for the app
+# Change user to non-root
 USER streamlit_user
 
-# Expose Streamlit default port
+# Expose Streamlit port
 EXPOSE 8501
 
-# Streamlit specific environment variable to disable browser opening
+# Environment settings for Streamlit
 ENV STREAMLIT_SERVER_HEADLESS=true
 
-# Run the Streamlit app
+# Run the app
 ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]