SloBERTa model, fine-tuned for natural language inference on Slovene dataset SI-NLI.
Usage
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("timkmecl/sloberta-sinli")
model = AutoModelForMaskedLM.from_pretrained("timkmecl/sloberta-sinli")
Expected inputs are of the form
Premisa: {premise}
Hipoteza: {hypothesis}
with strings {premise}
and {hypothesis}
being replaced with premise and hypothesis in Slovene.
Class 0 is entailment, class 1 neutral and 2 contradiction.