update Dockerfile

Dockerfile

@@ -7,7 +7,10 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 COPY . .
 
-# set huggingface cache directory to something writeable
+# Create writable cache directory
+RUN mkdir -p /app/.cache/huggingface && chmod -R 777 /app/.cache
+
+# Set huggingface cache directory
 ENV HF_HOME=/app/.cache/huggingface
 
 CMD ["python", "train.py"]