File size: 219 Bytes
9a62380
 
 
 
 
 
 
 
 
ef43fb2
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="cuda"
)