Update Dockerfile

Dockerfile

@@ -6,6 +6,8 @@ WORKDIR /app
 ENV HOME=/tmp
 ENV STREAMLIT_SERVER_HEADLESS=true
 ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
+ENV HUGGINGFACE_HUB_CACHE=/tmp/huggingface_cache
+ENV TRANSFORMERS_CACHE=/tmp/huggingface_cache
 
 RUN apt-get update && apt-get install -y \
     build-essential \
@@ -14,6 +16,9 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Create cache directory with proper permissions
+RUN mkdir -p /tmp/huggingface_cache && chmod 777 /tmp/huggingface_cache
+
 COPY requirements.txt ./
 COPY src/ ./src/