Fix: use /app/cache for Hugging Face models to avoid permission errors in Docker

Dockerfile

@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y \
 
 # Set Hugging Face cache to writable directory
 ENV TRANSFORMERS_CACHE=/app/cache
-RUN mkdir -p /app/cache
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
 
 # Copy requirements
 COPY requirements.txt .
@@ -27,5 +27,7 @@ COPY . .
 # Expose Uvicorn port
 EXPOSE 7860
 
+USER root
+
 # Start command
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

utils/mlp.py

@@ -1,3 +1,8 @@
+import os
+
+os.environ["TRANSFORMERS_CACHE"] = os.path.expanduser("~/.cache/huggingface")
+os.makedirs(os.environ["TRANSFORMERS_CACHE"], exist_ok=True)
+
 import torch
 import joblib
 from sentence_transformers import SentenceTransformer