luke-japanese-base-afinada-QA

Instale sentencepiece y transformers:

pip install sentencepiece
pip install transformers

Luego, ejecute el siguiente código para resolver tareas de Preguntas y Respuestas (Question-Answering):

import torch
from transformers import AutoTokenizer, LukeForQuestionAnswering

tokenizer = AutoTokenizer.from_pretrained('Mizuiro-sakura/luke-japanese-base-finetuned-QA')
model=LukeForQuestionAnswering.from_pretrained('Mizuiro-sakura/luke-japanese-base-finetuned-QA')

text={
'context':'私の名前はEIMIです。好きな食べ物は苺です。 趣味は皆さんと会話することです。',
'question' :'好きな食べ物は何ですか'
}

input_ids=tokenizer.encode(text['question'],text['context'])
output= model(torch.tensor([input_ids]))
prediction = tokenizer.decode(input_ids[torch.argmax(output.start_logits): torch.argmax(output.end_logits)])
print(prediction)