File size: 348 Bytes
9a7b546 | 1 2 3 4 5 6 7 8 9 10 11 12 | from sentence_transformers import SentenceTransformer
print("Downloading PubMedBERT embedding model...")
SentenceTransformer("NeuML/pubmedbert-base-embeddings")
print("Done.")
print("Downloading cross-encoder reranker model...")
SentenceTransformer("cross-encoder/ms-marco-MiniLM-L-6-v2")
print("Done.")
print("All models cached successfully.")
|