Update Dockerfile

Dockerfile

@@ -6,7 +6,9 @@ ENV PYTHONUNBUFFERED=1 \
     PORT=7860 \
     PYTHONPATH=/app \
     HOME=/home/chrome \
-    CHROME_BIN=/usr/bin/google-chrome
+    CHROME_BIN=/usr/bin/google-chrome \
+    TRANSFORMERS_CACHE=/app/.cache \
+    HF_HOME=/app/.cache
 
 # Create chrome user
 RUN mkdir -p /home/chrome && \
@@ -32,12 +34,16 @@ RUN apt-get update && apt-get install -y \
     && chown -R chrome:chrome /usr/bin/google-chrome \
     && rm -rf /var/lib/apt/lists/*
 
+# Create cache directory and set permissions
+RUN mkdir -p /app/.cache && chown -R chrome:chrome /app/.cache
+
 # Copy requirements file
 COPY requirements.txt .
 
-# Install Python dependencies
+# Install Python dependencies and pre-download the model
 RUN pip install --no-cache-dir -r requirements.txt \
-    && pip install --no-cache-dir hypercorn
+    && pip install --no-cache-dir hypercorn sentence-transformers \
+    && python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
 
 # Copy the current directory contents into the container
 COPY . /app