How to use kiel2/KielEmbed-Mini with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("kiel2/KielEmbed-Mini") sentences = [ "Name a style of hot yoga.", "Bikram.", "Tallahassee is the capital of Florida", "I want a redhead woman with tattoos and big boobs and a big ass" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
