Update Dockerfile

Dockerfile

@@ -11,7 +11,12 @@ RUN pip install --no-cache-dir transformers datasets accelerate peft torch torch
 COPY . .
 
 # Set a writable cache directory inside the app folder
+# Set writable cache directory inside the app folder
 ENV HF_HOME=/app/hf_cache
-RUN mkdir -p /app/hf_cache
+ENV HF_DATASETS_CACHE=/app/hf_cache
+ENV TRANSFORMERS_CACHE=/app/hf_cache
+
+# Create the cache directory and set correct permissions
+RUN mkdir -p /app/hf_cache/hub && chmod -R 777 /app/hf_cache
 # Set default command (replace with your training script later)
 CMD ["python", "train.py"]