1from sentence_transformers import SentenceTransformer
23# Download from the Hugging Face Hub4model = SentenceTransformer("sentence_transformers_model_id")5# Run inference6sentences =[7'search_query: autotrain',8'search_query: auto train',9'search_query: i love autotrain',10]11embeddings = model.encode(sentences)12print(embeddings.shape)1314# Get the similarity scores for the embeddings15similarities = model.similarity(embeddings, embeddings)16print(similarities.shape)
