Update Dockerfile

Dockerfile

@@ -2,6 +2,11 @@ FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
 
 ENV GRADIO_SERVER_PORT=7860
 ENV GRADIO_SERVER_NAME="0.0.0.0"
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV GRADIO_SERVER_PORT=7860
+ENV HF_HOME=/app/.cache/huggingface
 
 # Create mount point and set permissions for persistent storage
 RUN mkdir -p /mnt && \
@@ -17,6 +22,17 @@ COPY requirements.txt requirements.txt
 RUN apt-get update && apt-get install -y git && \
     pip install -r requirements.txt
 
+# Create a non-root user for better security
+RUN useradd -m -u 1000 appuser && \
+    mkdir -p /app/.cache/huggingface && \
+    mkdir -p /tmp/gradio_cache && \
+    mkdir -p /app/temp_audio && \
+    chown -R appuser:appuser /app && \
+    chown -R appuser:appuser /tmp/gradio_cache
+
+# Switch to non-root user
+USER appuser
+
 # Copy application files
 COPY app.py .
 COPY public/ /opt/app/public/