Update Dockerfile

Dockerfile

@@ -3,24 +3,36 @@ FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
 ENV DEBIAN_FRONTEND=noninteractive
 ENV TZ=Etc/UTC
 
-# Use /data as HF_HOME to avoid permission issues
-ENV HF_HOME=/data/huggingface
-
 WORKDIR /app
 
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     ffmpeg \
     git \
     tzdata \
     && rm -rf /var/lib/apt/lists/*
 
-# Create cache directory in writable location
-RUN mkdir -p /data/huggingface
+# Create a non-root user
+RUN useradd -m -u 1000 user
+
+# Create cache directory and set permissions
+RUN mkdir -p /data/huggingface/hub && \
+    chown -R user:user /data/huggingface
 
+# Set environment variables for HuggingFace cache
+ENV HF_HOME=/data/huggingface
+ENV TRANSFORMERS_CACHE=/data/huggingface/hub
+ENV HF_HUB_CACHE=/data/huggingface/hub
+
+# Copy requirements and install Python dependencies
 COPY requirements.txt .
 RUN pip install --upgrade pip && pip install -r requirements.txt
 
-COPY main.py .
+# Switch to non-root user
+USER user
+
+# Copy application code with proper ownership
+COPY --chown=user:user main.py .
 
 EXPOSE 7860