Update Dockerfile

Dockerfile

@@ -11,6 +11,9 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Install sentencepiece
+RUN pip install sentencepiece
+
 # Create the nltk_data directory and set permissions
 RUN mkdir -p /code/nltk_data && chmod -R 777 /code/nltk_data
 
@@ -20,10 +23,17 @@ ENV NLTK_DATA=/code/nltk_data
 # Download punkt data for NLTK
 RUN python -c "import nltk; nltk.download('punkt')"
 
+# Ensure the transformer cache directory exists
+RUN mkdir -p /code/transformers_cache
+ENV TRANSFORMERS_CACHE=/code/transformers_cache
+
+# Download the sentence-transformers model to avoid recreating it
+RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')"
+
 # Copy your code and models
 COPY . .
 
 # Run FastAPI
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
-ENV TRANSFORMERS_CACHE=/tmp/cache
+ENV TRANSFORMERS_CACHE=/code/transformers_cache