How to use swardiantara/bert-tiny-snli-k5-fixed-euclidean with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("swardiantara/bert-tiny-snli-k5-fixed-euclidean") sentences = [ "A woman in a green jacket and hood over her head looking towards a valley. [SEP] The woman is wearing green.", "A group of young children kick around a ball on a field with a body of water in the background. [SEP] The children are playing video games.", "A brown dog sits alone in front of a snowbank. [SEP] The dog is getting a tan on the beach.", "A man on stilts is playing a tuba for money on the boardwalk. [SEP] A male street performer with a tuba is playing outside." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
