distilbart-mnli-12-1-extended-labels

Para utilizar este modelo de clasificación de texto, puedes usar la biblioteca transformers de Hugging Face. Aquí tienes un fragmento de código de ejemplo:

from transformers import AutoTokenizer, AutoModelForSequenceClassification

# Cargar el tokenizador y el modelo
modelo = 'alramalho/distilbart-mnli-12-1-extended-labels'

# Crear una instancia del tokenizador
tokenizer = AutoTokenizer.from_pretrained(modelo)

# Crear una instancia del modelo
model = AutoModelForSequenceClassification.from_pretrained(modelo)

# Ejemplo de texto
texto = 'I like you. I love you'

# Tokenizar el texto
inputs = tokenizer(texto, return_tensors='pt')

# Realizar la clasificación
outputs = model(**inputs)

# Obtener los resultados de la clasificación
logits = outputs.logits