Update Dockerfile

Dockerfile

@@ -1,10 +1,6 @@
 FROM python:3.9-slim
 
-# Create cache directory with proper permissions
-RUN mkdir -p /app/.cache && \
-    chmod 777 /app/.cache
-
-# Install system dependencies
+# System dependencies
 RUN apt-get update && \
     apt-get install -y \
     gcc \
@@ -12,24 +8,29 @@ RUN apt-get update && \
     wkhtmltopdf \
     xvfb \
     fonts-liberation \
-    fonts-dejavu \
     && rm -rf /var/lib/apt/lists/*
 
+# Create cache directory with proper permissions
+RUN mkdir -p /app/.cache && \
+    chown -R 1000:1000 /app/.cache
+
 WORKDIR /app
 
-# Set environment variables for Hugging Face
+# Set environment variables
 ENV TRANSFORMERS_CACHE=/app/.cache \
     HF_DATASETS_CACHE=/app/.cache \
     XDG_CACHE_HOME=/app/.cache
 
-# Install Python dependencies
+# Install Python requirements
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt --mkdir
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Test model download with proper cache
+# Verify model download
 RUN python -c "from transformers import GPT2LMHeadModel; \
-GPT2LMHeadModel.from_pretrained('gpt2-medium', local_files_only=False)"
+GPT2LMHeadModel.from_pretrained('gpt2-medium', use_safetensors=True)"
 
 COPY . .
 
+# Run as non-root user
+USER 1000
 CMD ["gunicorn", "--bind", "0.0.0.0:5000", "--timeout", "600", "app:app"]