from sentence_transformers import SentenceTransformer, SimilarityFunction


model = SentenceTransformer(
    "sentence-transformers/all-MiniLM-L6-v2",
    similarity_fn_name=SimilarityFunction.COSINE,
)