Views
No views yet
1from sentence_transformers import CrossEncoder
2
3model = CrossEncoder('jacinthes/cross-encoder-sloberta-si-nli')
4premise = 'Pojdi z menoj v toplice.'
5hypothesis = 'Bova lepa bova fit.'
6prediction = model.predict([premise, hypothesis])
7int2label = {0: 'entailment', 1: 'neutral', 2:'contradiction'}
8print(int2label[prediction.argmax()])