Fix HF cache permission and correct model path, Fix DockerFile

Dockerfile

@@ -4,6 +4,12 @@ FROM python:3.10-slim
 # Set working directory
 WORKDIR /app
 
+
+# Set env for Hugging Face cache
+ENV HF_HOME=/tmp/huggingface
+ENV TRANSFORMERS_CACHE=/tmp/hf_cache
+ENV HF_HUB_CACHE=/tmp/hf_cache
+
 # Copy requirements first for caching
 COPY requirements.txt .
 

app/api/v1/metrics.py

@@ -6,13 +6,15 @@ from nltk.translate.bleu_score import sentence_bleu
 from sentence_transformers import SentenceTransformer, util
 from app.auth.auth import verify_token
 import os
-os.environ["HF_HOME"] = "/tmp/hf_cache"
+os.environ["HF_HOME"] = "/tmp/huggingface"
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
 os.environ["HF_HUB_CACHE"] = "/tmp/hf_cache"
 
 from sentence_transformers import SentenceTransformer
 
-model = SentenceTransformer("sentence-transformers/clip-ViT-B-32")
+model = SentenceTransformer("clip-ViT-B-32")  
+# OR multilingual version
+# model = SentenceTransformer("sentence-transformers/clip-ViT-B-32-multilingual-v1")32")
 
 router = APIRouter()