Update Dockerfile

Dockerfile

@@ -14,6 +14,9 @@ WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
+ENV TRANSFORMERS_CACHE=/code/.cache
+RUN mkdir -p /code/.cache && chmod -R 777 /code/.cache
+
 # 5. Install Python dependencies
 # Upgrade pip and install requirements, ensuring CPU PyTorch is used
 # Using --no-cache-dir makes the image smaller