Rustamshry commited on
Commit
2123dcb
·
verified ·
1 Parent(s): 5bfb063

Update src/embedding_utils.py

Browse files
Files changed (1) hide show
  1. src/embedding_utils.py +4 -4
src/embedding_utils.py CHANGED
@@ -1,15 +1,15 @@
1
  import os
2
  os.environ["SENTENCE_TRANSFORMERS_DISABLE_ONNX"] = "1"
3
 
4
- os.environ["TRANSFORMERS_CACHE"] = "/app/.cache"
5
- os.environ["HF_HOME"] = "/app/.cache"
6
- os.makedirs("/app/.cache", exist_ok=True)
7
 
8
  from sentence_transformers import SentenceTransformer
9
 
10
  model = SentenceTransformer(
11
  "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
12
- cache_folder="/app/.cache"
13
  )
14
 
15
 
 
1
  import os
2
  os.environ["SENTENCE_TRANSFORMERS_DISABLE_ONNX"] = "1"
3
 
4
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache"
5
+ os.environ["HF_HOME"] = "/tmp/.cache"
6
+ os.makedirs("/tmp/.cache", exist_ok=True)
7
 
8
  from sentence_transformers import SentenceTransformer
9
 
10
  model = SentenceTransformer(
11
  "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
12
+ cache_folder="/tmp/.cache"
13
  )
14
 
15