language: | |
- eu | |
# Basque Semantic Textual Similarity Model | |
```python | |
import transformers | |
if __name__ == '__main__': | |
model = 'eu-sts' | |
tokenizer = transformers.AutoTokenizer.from_pretrained(model) | |
model = transformers.BertForSequenceClassification.from_pretrained(model) | |
sentence_pairs = [("Nire ama etxera etorri da.", "Ama visitan etorri da etxera")] | |
for sp in sentence_pairs: | |
result = model(**tokenizer(*sp, return_tensors='pt')) | |
print(sp, result.logits) | |
``` |