updated Dockerfile with caches moved off the 50G root volume and onto /data

1 . New cache directories created on /data (Prevents hitting the 50 GB root limit)
2. Cache-cleaning step (Deletes any leftover Hugging Face caches during image build)
3.Updated environment variables (Moved from /tmp → /data so model/dataset downloads don’t fill root)

Dockerfile

@@ -11,14 +11,26 @@ RUN apt-get update && apt-get install -y \
     libomp-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Create writable directories
-RUN mkdir -p /tmp/huggingface_cache /tmp/matplotlib \
-    && chmod -R 777 /tmp/huggingface_cache /tmp/matplotlib
-
-# Set cache environment variables
-ENV HF_HOME=/tmp
-ENV TRANSFORMERS_CACHE=/tmp
-ENV MPLCONFIGDIR=/tmp
+# Create cache directories on /data and matplotlib config dir
+RUN mkdir -p \
+    /data/.huggingface \
+    /data/.cache/huggingface/datasets \
+    /data/.cache/huggingface/transformers \
+    /tmp/matplotlib \
+    && chmod -R 777 /data /tmp/matplotlib
+
+# Clean any old Hugging Face caches (defensive)
+RUN rm -rf \
+    /root/.cache/huggingface \
+    /root/.cache/huggingface_hub \
+    /root/.cache/huggingface/datasets || true
+
+# Set cache environment variables to point at /data
+ENV HF_HOME=/data/.huggingface
+ENV HF_HUB_CACHE=/data/.cache/huggingface/hub
+ENV HF_DATASETS_CACHE=/data/.cache/huggingface/datasets
+ENV TRANSFORMERS_CACHE=/data/.cache/huggingface/transformers
+ENV MPLCONFIGDIR=/tmp/matplotlib
 
 # Copy requirements file
 COPY requirements.txt /app/requirements.txt