monologg/koelectra-base-v2-finetuned-korquad-384

Para usar este modelo, se necesita cargar el modelo desde Hugging Face y aplicar la inferencia en los datos deseados. Un ejemplo de código para usar este modelo podría ser:

from transformers import ElectraForQuestionAnswering, ElectraTokenizer
import torch

model_name = 'monologg/koelectra-base-v2-finetuned-korquad-384'
model = ElectraForQuestionAnswering.from_pretrained(model_name)
tokenizer = ElectraTokenizer.from_pretrained(model_name)

question = "¿Cuál es la capital de Francia?"
context = "La capital de Francia es París."
inputs = tokenizer(question, context, return_tensors='pt')
start_scores, end_scores = model(**inputs).values()

all_tokens = tokenizer.convert_ids_to_tokens(inputs['input_ids'][0])
answer = ' '.join(all_tokens[torch.argmax(start_scores) : torch.argmax(end_scores)+1])
print(answer)