Update Dockerfile

Dockerfile

@@ -7,21 +7,26 @@ ENV PYTHONUNBUFFERED=1 \
     HF_HOME="/tmp/huggingface_cache" \
     HUGGINGFACE_HUB_CACHE="/tmp/huggingface_cache"
 
+# Create a non-root user
+RUN useradd --create-home --shell /bin/bash appuser
+
 # Set working directory
 WORKDIR /app
 
 # Copy requirements file
 COPY requirements.txt .  
 
-# Install dependencies using a virtual environment
+# Install dependencies and create virtual environment
 RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
 
 RUN python -m venv /app/venv && \
     /app/venv/bin/pip install --no-cache-dir --upgrade pip && \
     /app/venv/bin/pip install --no-cache-dir -r requirements.txt
 
-# Ensure the Hugging Face cache directory exists
-RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
+# Ensure the Hugging Face cache directory exists and assign proper permissions
+RUN mkdir -p $HF_HOME && \
+    chown -R appuser:appuser $HF_HOME && \
+    chmod -R 777 $HF_HOME
 
 # Copy application files
 COPY main.py .  
@@ -30,7 +35,9 @@ COPY main.py .
 ARG HF_TOKEN
 ENV HF_TOKEN=${HF_TOKEN}
 
-# Pre-download models
+# Pre-download models as the new user
+USER appuser
+
 RUN /app/venv/bin/python -c "from transformers import pipeline; \
     pipeline('sentiment-analysis', model='Ehrii/sentiment', use_auth_token='$HF_TOKEN')" || echo 'Failed to download multilingual model'