sentence-transformers How to use LamaDiab/MiniLM-V16Data-128BATCH-SemanticEngine with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("LamaDiab/MiniLM-V16Data-128BATCH-SemanticEngine")
sentences = [
"rosa / porcelain us andalusia mug",
"klara and the sun",
" mug",
"mug"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]