Update Dockerfile

Dockerfile

@@ -12,6 +12,7 @@ RUN npm install --frozen-lockfile
 COPY frontend/ ./
 RUN npm run build
 
+# --- Stage 2: GPU-enabled Python backend ---
 # --- Stage 2: GPU-enabled Python backend ---
 FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04 AS backend
 
@@ -22,6 +23,11 @@ ENV DEBIAN_FRONTEND=noninteractive \
     PIP_NO_CACHE_DIR=1 \
     HF_HOME=/app/.cache/huggingface
 
+# Install Python and system dependencies BEFORE switching user
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3 python3-pip python3-venv git curl && \
+    rm -rf /var/lib/apt/lists/*
+
 # Create non-root user
 RUN useradd -m appuser
 
@@ -32,13 +38,10 @@ RUN mkdir -p /app/.cache/huggingface && \
 # Switch to non-root user
 USER appuser
 
-# Install Python and system dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3 python3-pip python3-venv git curl && \
-    rm -rf /var/lib/apt/lists/*
-
 WORKDIR /app
 
+# Continue with pip installs and app setup...
+
 # Upgrade pip and install CUDA-enabled PyTorch
 RUN python3 -m pip install --upgrade pip && \
     python3 -m pip install --index-url https://download.pytorch.org/whl/cu121 torch==2.4.1+cu121