Update Dockerfile

Dockerfile

@@ -2,7 +2,7 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Set environment variables to fix permission issues
+# Set environment variables
 ENV HOME=/tmp
 ENV STREAMLIT_SERVER_HEADLESS=true
 ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
@@ -16,18 +16,21 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-# Create cache directory with proper permissions
+# Create cache directory
 RUN mkdir -p /tmp/huggingface_cache && chmod 777 /tmp/huggingface_cache
 
 COPY requirements.txt ./
+RUN pip3 install -r requirements.txt
+
+# Pre-download the model during build time
+RUN python -c "from sentence_transformers import SentenceTransformer; model = SentenceTransformer('BAAI/bge-small-en'); print('Model downloaded successfully')"
+
 COPY src/ ./src/
 
 # Create .streamlit directory and config file
 RUN mkdir -p /tmp/.streamlit
 RUN echo '[server]\nheadless = true\nport = 8501\nenableCORS = false\nenableXsrfProtection = false\n\n[browser]\ngatherUsageStats = false' > /tmp/.streamlit/config.toml
 
-RUN pip3 install -r requirements.txt
-
 EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health