Views
No views yet
1import transformers
2
3if __name__ == '__main__':
4 model = 'eu-sts'
5 tokenizer = transformers.AutoTokenizer.from_pretrained(model)
6 model = transformers.BertForSequenceClassification.from_pretrained(model)
7
8 sentence_pairs = [("Nire ama etxera etorri da.", "Ama visitan etorri da etxera")]
9 for sp in sentence_pairs:
10 result = model(**tokenizer(*sp, return_tensors='pt'))
11 print(sp, result.logits)
12