Update Dockerfile

Dockerfile

@@ -1,40 +1,35 @@
 FROM nvidia/cuda:12.6.0-runtime-ubuntu22.04
 
-RUN useradd -m -u 1000 user
-# System depsFROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
-
-RUN apt update && apt upgrade -y
-RUN apt-get -y install software-properties-common
-RUN add-apt-repository -y ppa:fish-shell/release-4
-RUN apt-get -y install git fish tmux
+RUN apt update && apt install -y \
+    git fish tmux curl ffmpeg \
+    libgl1-mesa-glx libglib2.0-0 \
+    build-essential ninja-build python3.10-venv
 
-RUN apt-get -y install curl
-RUN apt-get install -y ffmpeg
+# Create user
+RUN useradd -m -u 1000 user
+USER user
+WORKDIR /home/user
 
-RUN apt-get install -y libgl1-mesa-glx
-RUN apt-get install -y libglib2.0-0
-RUN apt-get install -y build-essential
-RUN apt-get install -y ninja-build
-RUN apt-get install -y python3.10-venv
+# Create HF cache with correct ownership
+ENV HF_HOME=/home/user/.cache/huggingface
+RUN mkdir -p $HF_HOME
 
-ENV TRANSFORMERS_CACHE=/home/user/.cache/huggingface
-RUN mkdir -p /home/user/.cache/huggingface
+# Thread settings
 ENV OMP_NUM_THREADS=1
+ENV MKL_NUM_THREADS=1
+ENV NUMEXPR_NUM_THREADS=1
 
 # Create virtualenv
-RUN python3 -m venv /opt/venv
-ENV PATH="/opt/venv/bin:$PATH"
+RUN python3 -m venv /home/user/venv
+ENV PATH="/home/user/venv/bin:$PATH"
 
+# Dependencies
 COPY --chown=user requirements.txt .
 RUN pip install --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-# Copy rest of app source
+# Copy app
 COPY --chown=user . .
 
-# Switch to non-root
-USER user
-
 EXPOSE 7860
-
 CMD ["python3", "app.py"]