Spaces:
Sleeping
Sleeping
Update src/embedding_utils.py
Browse files- 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"] = "/
|
| 5 |
-
os.environ["HF_HOME"] = "/
|
| 6 |
-
os.makedirs("/
|
| 7 |
|
| 8 |
from sentence_transformers import SentenceTransformer
|
| 9 |
|
| 10 |
model = SentenceTransformer(
|
| 11 |
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
|
| 12 |
-
cache_folder="/
|
| 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 |
|