Update Dockerfile

Dockerfile

@@ -13,6 +13,17 @@ RUN apt-get update && apt-get install -y \
     python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
+# Create cache directories with proper permissions
+RUN mkdir -p /app_cache/huggingface \
+    mkdir -p /app_cache/matplotlib \
+    chmod -R 777 /app_cache
+
+# Set environment variables to use custom cache locations
+ENV TRANSFORMERS_CACHE=/app_cache/huggingface
+ENV HF_HOME=/app_cache/huggingface
+ENV MPLCONFIGDIR=/app_cache/matplotlib
+ENV TORCH_HOME=/app_cache/torch
+
 # Upgrade pip
 RUN pip install --upgrade pip
 
@@ -26,10 +37,8 @@ COPY . .
 # Build Cython extensions
 RUN cd utils && python setup.py build_ext --inplace
 
-# Create necessary directories if they don't exist and fix permissions
+# Create necessary directories if they don't exist
 RUN mkdir -p data/models && \
-    mkdir -p .cache && \
-    chmod -R 777 .cache && \
     chmod -R 777 data
 
 # Expose the port for FastAPI application