Spaces:
Sleeping
Sleeping
Commit
•
932acc8
1
Parent(s):
8298f51
Create download_model.py
Browse files- download_model.py +6 -0
download_model.py
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
from InstructorEmbedding import INSTRUCTOR
|
| 3 |
+
|
| 4 |
+
CACHE_DIR = "/app/src/fos_mapper/data/model_artefacts"
|
| 5 |
+
|
| 6 |
+
model = INSTRUCTOR(embedding_model, cache_folder=CACHE_DIR, device='cpu')
|