Update Dockerfile

Dockerfile

@@ -22,6 +22,9 @@ RUN apt-get update && apt-get install -y \
 # Set working directory
 WORKDIR /app
 
+# Create cache directory for Hugging Face
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+
 # Copy application files
 COPY . /app
 
@@ -35,6 +38,8 @@ ENV CHROMIUM_PATH=/usr/bin/chromium
 ENV DISPLAY=:99
 ENV PYTHONUNBUFFERED=1
 ENV STREAMLIT_OPT_OUT_OF_USAGE_STATISTICS=true
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HOME=/app/cache
 
 # Ensure ChromeDriver and Chromium are executable
 RUN chmod +x /usr/bin/chromedriver /usr/bin/chromium
@@ -43,4 +48,4 @@ RUN chmod +x /usr/bin/chromedriver /usr/bin/chromium
 EXPOSE 8501
 
 # Run Streamlit with headless mode and disabled file watcher
-CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.fileWatcherType=none", "--server.headless=true"]
+CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.fileWatcherType=none", "--server.headless=true"]