Update Dockerfile

Dockerfile

@@ -1,26 +1,28 @@
-# Use official Python image
 FROM python:3.9-slim
 
-# Install system dependencies
+# Install system deps
 RUN apt-get update && apt-get install -y \
     ffmpeg \
     libsndfile1 \
     && rm -rf /var/lib/apt/lists/*
 
-# Set working directory
+# Create non-root user and cache directory
+RUN useradd -m appuser && \
+    mkdir -p /app/.cache/huggingface && \
+    chown -R appuser:appuser /app
+
+# Set environment variables for Hugging Face cache
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface \
+    HF_DATASETS_CACHE=/app/.cache/huggingface
+
+# Switch to non-root user
+USER appuser
+
 WORKDIR /app
 
-# Copy requirements first to leverage Docker cache
 COPY requirements.txt .
-
-# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application code
 COPY . .
 
-# Expose port
-EXPOSE 8000
-
-# Start application
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]