一站式网上办事大厅

import torch

from transformers import AutoTokenizer, AutoModelForQuestionAnswering

tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")

model = AutoModelForQuestionAnswering.from_pretrained("bert-base-uncased")

question = "如何申请助学金？"

context = "学生可以通过学校官网的‘一网通办’平台提交助学金申请材料。"

inputs = tokenizer.encode_plus(question, context, return_tensors="pt")

outputs = model(**inputs)

answer_start = torch.argmax(outputs.start_logits)

answer_end = torch.argmax(outputs.end_logits) + 1

answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs["input_ids"][0][answer_start:answer_end]))

print("答案：", answer)