timkmecl/sloberta-sinli

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.