Update Dockerfile

Dockerfile

@@ -1,24 +1,23 @@
-FROM python:3.10-slim
-
+# Set working directory
 WORKDIR /app
 
 # Install system packages
 RUN apt-get update && apt-get install -y ffmpeg git wget && rm -rf /var/lib/apt/lists/*
 
-# Defined permission for pretrained_models
+# Set environment for proper model caching and permissions
 ENV PYTHONUNBUFFERED=1
-RUN mkdir -p /app/pretrained_models && chmod -R 777 /app/pretrained_models
 ENV HOME=/app
-
-# Set cache envs (for Whisper + Hugging Face + SpeechBrain)
 ENV XDG_CACHE_HOME=/app/.cache
 ENV SPEECHBRAIN_CACHE=/app/.cache/speechbrain
 
-# Create cache dirs with write permission
-RUN mkdir -p /app/.cache/whisper /app/.cache/speechbrain && chmod -R 777 /app/.cache
+# Create model/cache directories with correct permissions
+RUN mkdir -p /app/pretrained_models \
+    && mkdir -p /app/wav2vec2_checkpoints \
+    && mkdir -p /app/.cache/whisper /app/.cache/speechbrain \
+    && chmod -R 777 /app/pretrained_models /app/wav2vec2_checkpoints /app/.cache
 
 # Upgrade pip
-RUN pip install --no-cache-dir --upgrade pip
+RUN pip install --no-cache-dir --upgrade pip --root-user-action=ignore
 
 # Copy source
 COPY . .
@@ -27,7 +26,7 @@ COPY . .
 RUN mkdir -p src && wget -O src/custome_interface.py https://huggingface.co/Jzuluaga/accent-id-commonaccent_xlsr-en-english/resolve/main/custom_interface.py
 
 # Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt --root-user-action=ignore
 
 # Expose Streamlit port
 EXPOSE 8501