Cómo usar

from transformers import XLMRobertaTokenizerFast, XLMRobertaForSequenceClassification

# cargar el tokenizador y los pesos del modelo
tokenizer = XLMRobertaTokenizerFast.from_pretrained('s-nlp/xlmr_formality_classifier')
model = XLMRobertaForSequenceClassification.from_pretrained('s-nlp/xlmr_formality_classifier')

id2formality = {0: 'formal', 1: 'informal'}
texts = [
"Me gustas. Te quiero.",
"Hey, ¿qué pasa?",
"Siema, ¿qué haces?",
"Siento un profundo pesar y tristeza por la situación en la política internacional.",
]

# preparar la entrada
encoding = tokenizer(
texts,
add_special_tokens=True,
return_token_type_ids=True,
truncation=True,
padding='max_length',
return_tensors='pt',
)

# inferencia
output = model(**encoding)

formality_scores = [
{id2formality[idx]: score for idx, score in enumerate(text_scores.tolist())}
for text_scores in output.logits.softmax(dim=1)
]
formality_scores