scinobo-fos-classification / download_model.py
Last commit not found
raw
history blame
218 Bytes
# 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"
)