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.")