A new cache directory

Dockerfile

@@ -11,18 +11,16 @@ RUN apt-get update && apt-get install -y python3 python3-pip \
 RUN pip install --upgrade pip
 RUN pip install --upgrade git+https://github.com/huggingface/transformers.git accelerate datasets[audio]
 
-RUN chmod -R 777 ~/.cache
-
 # 
 WORKDIR /code
 
+RUN mkdir -p /app/.cache
+RUN chmod -R 777 /app/.cache
+ENV HUGGINGFACE_CACHE_DIR=/app/.cache
+
 # 
 COPY ./requirements.txt /code/requirements.txt
 
-# Download the models
-RUN python3 -c 'from transformers import AutoModelForSpeechSeq2Seq; AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-large-v3");'
-RUN python3 -c 'from transformers import AutoProcessor; AutoProcessor.from_pretrained("openai/whisper-large-v3");'
-
 # 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt