File size: 218 Bytes
9a62380
 
 
 
 
 
 
 
 
218e814
9a62380
1
2
3
4
5
6
7
8
9
10
11
# download_model.py
from sentence_transformers import SentenceTransformer
import os

CACHE_DIR = "/app/src/fos/data"

model = SentenceTransformer(
    "all-mpnet-base-v2",
    cache_folder=CACHE_DIR,
    device="cpu"
)