How to use OussamaBenSlama/e5-small-dialect2MSA with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("OussamaBenSlama/e5-small-dialect2MSA") sentences = [ "هذا شخص سعيد", "هذا كلب سعيد", "هذا شخص سعيد جدا", "اليوم هو يوم مشمس" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
