sentence-transformers How to use tomaarsen/mmBERT-small-msmarco with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("tomaarsen/mmBERT-small-msmarco")
sentences = [
"what county is lyndhurst, ohio in",
"This article is about the song written by Kenneth Gamble, Leon Huff and Cary Gilbert. For the Tina Turner song, see Don't Leave Me This Way (Tina Turner song). Don't Leave Me This Way is a song written by Kenneth Gamble, Leon Huff and Cary Gilbert. First charting as a hit for Harold Melvin & the Blue Notes featuring Teddy Pendergrass, an act on Gamble & Huff's Philadelphia International label in 1975, Don't Leave Me This Way was later a huge disco hit for Motown artist Thelma Houston in 1977.",
"Lyndhurst is a city in Cuyahoga County, Ohio, United States. The population was 14,001 at the 2010 census. Lyndhurst is located in northeastern Ohio, and is a suburb of Cleveland. A small part of Lyndhurst was originally part of Mayfield Township. It used to be called Euclidville before Lyndhurst was chosen. Lyndhurst is located at 41°31â²17â³N 81°29â²25â³W / 41.52139°N 81.49028°W / 41.52139; -81.49028 (41.521352, -81.490141).",
"Welcome to Trumbull County... Trumbull County, the county seat, located in Warren, Ohio, consists of a combination of both urban and rural communities situated in the northeast corner of Ohio. It is situated roughly between the Youngstown, Cleveland and Akron corridors."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
