osnarayana commited on
Commit
6693f3f
·
1 Parent(s): fa9dd97

Fix HF cache permission and correct model path

Browse files
Files changed (1) hide show
  1. app/api/v1/metrics.py +8 -1
app/api/v1/metrics.py CHANGED
@@ -5,9 +5,16 @@ from sklearn.metrics import accuracy_score
5
  from nltk.translate.bleu_score import sentence_bleu
6
  from sentence_transformers import SentenceTransformer, util
7
  from app.auth.auth import verify_token
 
 
 
 
 
 
 
 
8
 
9
  router = APIRouter()
10
- model = SentenceTransformer("clip-ViT-B-32") # for CLIP-like semantic score
11
 
12
  @router.post("/evaluate/bleu", dependencies=[Depends(verify_token)])
13
  def compute_bleu(reference: str, candidate: str):
 
5
  from nltk.translate.bleu_score import sentence_bleu
6
  from sentence_transformers import SentenceTransformer, util
7
  from app.auth.auth import verify_token
8
+ import os
9
+ os.environ["HF_HOME"] = "/tmp/hf_cache"
10
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
11
+ os.environ["HF_HUB_CACHE"] = "/tmp/hf_cache"
12
+
13
+ from sentence_transformers import SentenceTransformer
14
+
15
+ model = SentenceTransformer("sentence-transformers/clip-ViT-B-32")
16
 
17
  router = APIRouter()
 
18
 
19
  @router.post("/evaluate/bleu", dependencies=[Depends(verify_token)])
20
  def compute_bleu(reference: str, candidate: str):