Update Dockerfile

Dockerfile

@@ -1,15 +1,13 @@
-# Use official Python image
 FROM python:3.10
 
-# Set working directory
 WORKDIR /app
 
-# Copy requirements and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of the code
 COPY . .
 
-# Run the training script
+# set huggingface cache directory to something writeable
+ENV HF_HOME=/app/.cache/huggingface
+
 CMD ["python", "train.py"]