Update Dockerfile

update cache write permission

Dockerfile

@@ -2,6 +2,7 @@ FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV TZ=Etc/UTC
+ENV TRANSFORMERS_CACHE=/app/cache
 
 WORKDIR /app
 
@@ -11,6 +12,8 @@ RUN apt-get update && apt-get install -y \
     tzdata \
     && rm -rf /var/lib/apt/lists/*
 
+RUN mkdir -p /app/cache
+
 COPY requirements.txt .
 RUN pip install --upgrade pip && pip install -r requirements.txt