How to use PartAI/Tooka-SBERT with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("PartAI/Tooka-SBERT") sentences = [ "درنا از پرندگان مهاجر با پاهای بلند و گردن دراز است.", "درناها با قامتی بلند و بالهای پهن، از زیباترین پرندگان مهاجر به شمار میروند.", "درناها پرندگانی کوچک با پاهای کوتاه هستند که مهاجرت نمیکنند.", "ایران برای بار دیگر توانست به مدال طلا دست یابد." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
