Update Dockerfile

Dockerfile

@@ -5,10 +5,10 @@ FROM python:3.10-slim
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 
-# Set working directory
+# Set working directory to home/streamlit_user (non-root user directory)
 WORKDIR /home/streamlit_user/app
 
-# Install system dependencies
+# Install system dependencies (including for image processing and huggingface)
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     git \
@@ -16,7 +16,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libgl1-mesa-glx \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first to leverage Docker cache
+# Create a directory for model weights under the user directory and set permissions
+RUN mkdir -p /home/streamlit_user/step1x_weights && \
+    chown -R streamlit_user:streamlit_user /home/streamlit_user/step1x_weights
+
+# Copy requirements.txt first to leverage Docker cache
 COPY requirements.txt ./requirements.txt
 
 # Install Python dependencies
@@ -26,6 +30,9 @@ RUN pip install --upgrade pip && \
 # Copy the rest of your app files
 COPY src/ ./src/
 
+# Set user for the app
+USER streamlit_user
+
 # Expose Streamlit default port
 EXPOSE 8501