update dockerfile

Dockerfile

@@ -8,6 +8,17 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 # Set the working directory in the container
 WORKDIR /app
 
+# --- Define cache directory early for clarity ---
+ARG HF_CACHE_DIR_ARG=/app/.cache
+ENV HF_HOME=${HF_CACHE_DIR_ARG}
+ENV HF_DATASETS_CACHE=${HF_CACHE_DIR_ARG}/datasets
+ENV TRANSFORMERS_CACHE=${HF_CACHE_DIR_ARG}/transformers
+# Add other HF cache variables if needed (e.g., HUGGINGFACE_HUB_CACHE)
+
+# Create the cache directories and make them writable
+# This also creates parent directories like /app/.cache
+RUN mkdir -p ${HF_DATASETS_CACHE} && chmod -R 777 ${HF_CACHE_DIR_ARG}
+
 COPY setup.sh .
 COPY requirements.txt .
 COPY app.py .