Update Dockerfile

Dockerfile

@@ -23,12 +23,13 @@ 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
+# Ensure the transformer cache directory exists and set correct permissions
+RUN mkdir -p /code/transformers_cache && chmod -R 777 /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')"
+RUN python -c "from sentence_transformers import CrossEncoder; CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')"
 
 # Copy your code and models
 COPY . .