How to use swardiantara/bert-tiny-snli-k3-fixed-cosine with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("swardiantara/bert-tiny-snli-k3-fixed-cosine") sentences = [ "A woman is walking across the street eating a banana, while a man is following with his briefcase. [SEP] The woman and man are playing baseball together.", "A woman with a panda hat and headphones is in front of a man outside in the snow. [SEP] The man is on the beach.", "A woman with a panda hat and headphones is in front of a man outside in the snow. [SEP] The man is on the beach.", "Male standing in the roadway, he is wearing a light colored shirt and holding a backpack. [SEP] A male is watching his dogs play." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
The community tab is the place to discuss and collaborate with the HF community!
