TunBERT

Cómo cargar el modelo

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("not-lain/TunBERT")
model = AutoModelForSequenceClassification.from_pretrained("not-lain/TunBERT",trust_remote_code=True)

Cómo usar el modelo

text = "[insert text here]"
inputs = tokenizer(text, return_tensors='pt')
output = model(**inputs)

O puedes usar el pipeline:

from transformers import pipeline

pipe = pipeline(model="not-lain/TunBERT",tokenizer = "not-lain/TunBERT",trust_remote_code=True)
pipe("text")

IMPORTANTE: Asegúrate de habilitar trust_remote_code=True