Update Dockerfile

Dockerfile

@@ -10,7 +10,9 @@ ENV PYTHONUNBUFFERED=1 \
     HF_HOME=/tmp/huggingface \
     TRANSFORMERS_CACHE=/tmp/huggingface \
     HF_DATASETS_CACHE=/tmp/huggingface \
-    HF_HUB_CACHE=/tmp/huggingface
+    HF_HUB_CACHE=/tmp/huggingface \
+    TORCH_HOME=/tmp/huggingface \
+    XDG_CACHE_HOME=/tmp/huggingface
 
 # ===== System Dependencies =====
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -25,7 +27,9 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # ===== Cache & Permissions =====
-RUN mkdir -p /tmp/huggingface && chmod -R 777 /tmp/huggingface
+RUN rm -rf /.cache /root/.cache /root/.huggingface && \
+    mkdir -p /tmp/huggingface && \
+    chmod -R 777 /tmp /app
 
 # ===== Copy Project Files =====
 COPY . .