Update Dockerfile

Dockerfile

@@ -7,6 +7,12 @@ RUN --mount=type=secret,id=huggingface_token,mode=0444,required=true \
  git init && \
  git remote add origin $(cat /run/secrets/huggingface_token)
 
+# Set the TRANSFORMERS_CACHE environment variable
+ENV TRANSFORMERS_CACHE=/tmp/.cache/huggingface
+
+# Create the cache folder with appropriate permissions
+RUN mkdir -p $TRANSFORMERS_CACHE && chmod -R 777 $TRANSFORMERS_CACHE
+
 
 COPY ./requirements.txt /code/requirements.txt