Update Dockerfile

Dockerfile

@@ -5,11 +5,15 @@ RUN apt update && \
     apt clean
 
 # Set Hugging Face and model cache dirs to a writable location
-ENV HF_HOME=/tmp/.cache/huggingface
-ENV TRANSFORMERS_CACHE=/tmp/.cache/huggingface
-ENV XDG_CACHE_HOME=/tmp/.cache
+ENV HF_HOME=/data/hf
+ENV TRANSFORMERS_CACHE=/data/hf
+ENV XDG_CACHE_HOME=/data/hf
 ENV MPLCONFIGDIR=/tmp/matplotlib
 
+RUN mkdir -p /data/hf && chmod -R 777 /data/hf
+
+RUN python3 -c "from faster_whisper import WhisperModel; WhisperModel('base', compute_type='int8')"
+
 WORKDIR /app
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt