sentence-transformers 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]