personalisation-engine / scripts /download_model.py
nice-bill's picture
Readme updated
2d773b1
raw
history blame contribute delete
254 Bytes
from sentence_transformers import SentenceTransformer
import os
MODEL_NAME = "all-MiniLM-L6-v2"
def download():
print(f"Downloading {MODEL_NAME}...")
SentenceTransformer(MODEL_NAME)
print("Done.")
if __name__ == "__main__":
download()