Update Dockerfile

Dockerfile

@@ -5,19 +5,20 @@ WORKDIR /app
 # Install system packages
 RUN apt-get update && apt-get install -y ffmpeg git wget && rm -rf /var/lib/apt/lists/*
 
-# Set cache env (for Whisper and Hugging Face)
+# 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 && chmod -R 777 /app/.cache
+RUN mkdir -p /app/.cache/whisper /app/.cache/speechbrain && chmod -R 777 /app/.cache
 
 # Upgrade pip
 RUN pip install --no-cache-dir --upgrade pip
 
-# Copy source code
+# Copy source
 COPY . .
 
-# Download Hugging Face custom interface file
+# Download Hugging Face interface file
 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