Update Dockerfile

Dockerfile

@@ -2,6 +2,8 @@ FROM python:3.11-slim
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
+ENV TRANSFORMERS_CACHE=/app/.cache
+ENV HF_HOME=/app/.cache
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -26,7 +28,7 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Install additional Hugging Face and related libraries with explicit updates
+# Install additional libraries
 RUN pip install -U \
     transformers \
     sentence-transformers \
@@ -41,13 +43,23 @@ RUN pip install playwright && playwright install --with-deps chromium
 COPY . /app
 WORKDIR /app
 
-# Set appropriate permissions
+# Create cache directory and set permissions
+RUN mkdir -p /app/.cache && \
+    chmod -R 777 /app/.cache
+
+# Set appropriate permissions for the entire app
 RUN chmod -R 755 /app
 
-# Preload SentenceTransformer model with error handling and trust remote code
-RUN python -c "from sentence_transformers import SentenceTransformer; from transformers import AutoModel, AutoTokenizer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', trust_remote_code=True)"
+# Preload SentenceTransformer model with explicit caching
+RUN python -c "
+import os
+os.environ['TRANSFORMERS_CACHE'] = '/app/.cache'
+os.environ['HF_HOME'] = '/app/.cache'
+from sentence_transformers import SentenceTransformer
+model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', cache_folder='/app/.cache')
+"
 
-# Expose port (optional, for web apps)
+# Expose port
 EXPOSE 7860
 
 # Run the app