umarigan/all_nli_tr
Viewer • Updated • 942k • 122
How to use umarigan/distilbert-turkish-sentence-similarity with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("umarigan/distilbert-turkish-sentence-similarity")
model = AutoModel.from_pretrained("umarigan/distilbert-turkish-sentence-similarity")pip install -U sentence-transformers
Load the model and run inference.
from sentence_transformers import SentenceTransformer
# Download from the 🤗 Hub
model = SentenceTransformer("umarigan/distilbert-turkish-sentence-similarity")
# Run inference
sentences = [
'Bu yıl tatile Antalya'ya gideceğiz.',
'Yazın Antalya'da tatil yapacağız.'
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [2, 768]
# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [2, 2]