scinobo-fos-classification / download_model.py
sotiriskot's picture
Update download_model.py
218e814 verified
# 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"
)