Update Dockerfile

Dockerfile

@@ -10,5 +10,7 @@ RUN pip install --no-cache-dir transformers datasets accelerate peft torch torch
 # Copy all local files into the container
 COPY . .
 
+ENV TRANSFORMERS_CACHE=/app/cache
+RUN mkdir -p /app/cache
 # Set default command (replace with your training script later)
 CMD ["python", "train.py"]