Update Dockerfile

Dockerfile

@@ -19,6 +19,15 @@ RUN pip install --no-cache-dir --upgrade pip && \
 # Copy the rest of the application code into the container
 COPY . .
 
+# ---- ADD THIS LINE ----
+# Set environment variables for Hugging Face cache directories
+# Use directories within /app where the user has write permissions.
+ENV HF_HOME=/app/.cache/huggingface
+ENV HF_DATASETS_CACHE=/app/.cache/datasets
+# Alternatively, just HF_DATASETS_CACHE might be enough if only datasets are used.
+# Using HF_HOME covers models, datasets, etc.
+# ---- END OF ADDED LINE ----
+
 # Expose the port the app runs on
 # This informs Docker that the container listens on this port
 EXPOSE 7860