Update Dockerfile

Dockerfile

@@ -1,8 +1,7 @@
-# Use a slim base image
+# Base image
 FROM python:3.10-slim
 
-# Install system dependencies
-RUN apt-get update && \
+# Install system dependencies\ nRUN apt-get update && \
     apt-get install -y git && \
     rm -rf /var/lib/apt/lists/*
 
@@ -10,19 +9,22 @@ RUN apt-get update && \
 COPY requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir -r /app/requirements.txt
 
-# Create and set working directory
+# Set working directory
 WORKDIR /app
 COPY app.py /app/app.py
 
-# Create a writable cache directory
-RUN mkdir -p /app/cache
+# Create and permit cache directory
+RUN mkdir -p /app/cache && \
+    chmod -R a+rwx /app/cache
 
-# Override default cache paths
-ENV TRANSFORMERS_CACHE=/app/cache
+# Override cache paths
 ENV HF_HOME=/app/cache
+ENV HF_DATASETS_CACHE=/app/cache
+# Transformer's own cache var is deprecated but safe to set
+ENV TRANSFORMERS_CACHE=/app/cache
 
-# Expose port for your UI
+# Expose port (Gradio/Streamlit default)
 EXPOSE 7860
 
-# Run the application
+# Launch
 CMD ["python", "app.py"]