Nidhi-Phophaliya commited on
Commit
8cdd015
·
verified ·
1 Parent(s): 4fa995b

Update app/utils/embedder.py

Browse files
Files changed (1) hide show
  1. app/utils/embedder.py +4 -3
app/utils/embedder.py CHANGED
@@ -4,14 +4,15 @@ import gdown
4
  import os
5
  import faiss
6
  import numpy as np
 
7
  import pickle
8
  from sentence_transformers import SentenceTransformer
9
 
10
  class Embedder:
11
  def __init__(self, model_name='sentence-transformers/all-MiniLM-L6-v2'):
12
- os.environ['TRANSFORMERS_CACHE'] = './cache'
13
- os.environ['HF_HOME'] = './cache'
14
- os.makedirs('./cache', exist_ok=True)
15
 
16
  self.model = SentenceTransformer(model_name)
17
  self.index = None
 
4
  import os
5
  import faiss
6
  import numpy as np
7
+ import tempfile
8
  import pickle
9
  from sentence_transformers import SentenceTransformer
10
 
11
  class Embedder:
12
  def __init__(self, model_name='sentence-transformers/all-MiniLM-L6-v2'):
13
+ cache_dir = tempfile.gettempdir()
14
+ os.environ['TRANSFORMERS_CACHE'] = cache_dir
15
+ os.environ['HF_HOME'] = cache_dir
16
 
17
  self.model = SentenceTransformer(model_name)
18
  self.index = None