Update Dockerfile

Dockerfile

@@ -12,7 +12,11 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt ./
 COPY src/ ./src/
 
+# Hugging Face cache config
 ENV TRANSFORMERS_CACHE=/app/cache/huggingface
+ENV HF_HOME=/app/cache/huggingface
+ENV HUGGINGFACE_HUB_CACHE=/app/cache/huggingface
+
 RUN mkdir -p /app/cache/huggingface && chmod -R 777 /app/cache
 
 RUN pip3 install -r requirements.txt
@@ -21,4 +25,4 @@ 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"]