Update Dockerfile

Dockerfile

@@ -10,6 +10,9 @@ RUN useradd -m -u 1000 appuser
 # Create cache directory and set permissions
 RUN mkdir -p /app/cache && chown -R appuser:appuser /app
 
+# Set environment variable to disable Xet Storage early
+ENV HUGGINGFACE_HUB_DISABLE_XET=true
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
@@ -19,15 +22,24 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Pre-download models to cache with retry mechanism
+RUN for i in 1 2 3 4 5; do \
+    python -c "from langchain_huggingface import HuggingFaceEmbeddings; HuggingFaceEmbeddings(model_name='all-MiniLM-L6-v2')" && break || \
+    echo "Retry $i failed for all-MiniLM-L6-v2, waiting before next attempt..." && sleep 10; \
+    done || { echo "Failed to download all-MiniLM-L6-v2 after 5 retries"; exit 1; }
+RUN for i in 1 2 3 4 5; do \
+    python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; AutoTokenizer.from_pretrained('distilgpt2'); AutoModelForCausalLM.from_pretrained('distilgpt2')" && break || \
+    echo "Retry $i failed for distilgpt2, waiting before next attempt..." && sleep 10; \
+    done || { echo "Failed to download distilgpt2 after 5 retries"; exit 1; }
+
 # Copy all project files
 COPY . .
 
 # Switch to non-root user
 USER appuser
 
-# Set environment variables for Hugging Face cache and disable Xet Storage
+# Set environment variable for Hugging Face cache
 ENV HF_HOME=/app/cache
-ENV HUGGINGFACE_HUB_DISABLE_XET=true
 
 # Expose the Flask port
 EXPOSE 8000