kobigbird-pure23-34112365

Para usar el modelo, puede seguir el siguiente ejemplo de código en Python usando la biblioteca Transformers:

from transformers import AutoTokenizer, AutoModelForQuestionAnswering

# Cargar el modelo y el tokenizador
model_name = 'sudong97/kobigbird-pure23-34112365'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForQuestionAnswering.from_pretrained(model_name)

# Definir el contexto y la pregunta
context = "Mi nombre es Clara y vivo en Berkeley."
question = "Cuál es mi nombre?"

# Tokenizar la entrada
inputs = tokenizer(question, context, return_tensors='pt')

# Obtener la respuesta
outputs = model(**inputs)

# Extraer la respuesta
start_scores = outputs.start_logits
end_scores = outputs.end_logits
all_tokens = tokenizer.convert_ids_to_tokens(inputs['input_ids'].numpy()[0])
answer = ' '.join(all_tokens[torch.argmax(start_scores) : torch.argmax(end_scores)+1])
print(answer)