Update Dockerfile

Dockerfile

@@ -14,8 +14,16 @@ COPY src/ ./src/
 
 RUN pip3 install -r requirements.txt
 
+# Set cache directory to avoid permission issues during model download
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HOME=/app/cache
+ENV SENTENCE_TRANSFORMERS_HOME=/app/cache
+
+# Create the cache directory
+RUN mkdir -p /app/cache
+
 EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]