align_to_words=False

README.md

@@ -28,17 +28,11 @@ This is a DeBERTa(V2) model pretrained on Japanese Wikipedia and 青空文庫 te
 ## How to Use
 
 ```py
-import torch
-from transformers import AutoTokenizer,AutoModelForQuestionAnswering
+from transformers import AutoTokenizer,AutoModelForQuestionAnswering,QuestionAnsweringPipeline
 tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/deberta-base-japanese-wikipedia-ud-head")
 model=AutoModelForQuestionAnswering.from_pretrained("KoichiYasuoka/deberta-base-japanese-wikipedia-ud-head")
-question="国語"
-context="全学年にわたって小学校の国語の教科書に挿し絵が用いられている"
-inputs=tokenizer(question,context,return_tensors="pt",return_offsets_mapping=True)
-offsets=inputs.pop("offset_mapping").tolist()[0]
-outputs=model(**inputs)
-start,end=torch.argmax(outputs.start_logits),torch.argmax(outputs.end_logits)
-print(context[offsets[start][0]:offsets[end][-1]])
+qap=QuestionAnsweringPipeline(tokenizer=tokenizer,model=model,align_to_words=False)
+print(qap(question="国語",context="全学年にわたって小学校の国語の教科書に挿し絵>が用いられている"))
 ```
 
 or (with [ufal.chu-liu-edmonds](https://pypi.org/project/ufal.chu-liu-edmonds/))