def process_request(request_type):
if request_type == "学籍证明":
return "https://school.edu/academic_certificate"
elif request_type == "请假":
return "https://school.edu/leave_application"
else:
return "未知请求类型，请联系管理员。"
]]>
import torch
from transformers import BertTokenizer, BertForQuestionAnswering
tokenizer = BertTokenizer.from_pretrained('bert-base-chinese')
model = BertForQuestionAnswering.from_pretrained('bert-large-uncased-whole-word-masking-finetuned-squad')
def answer_question(question, context):
inputs = tokenizer.encode_plus(question, context, add_special_tokens=True, return_tensors="pt")
outputs = model(**inputs)
answer_start = torch.argmax(outputs.start_logits)
answer_end = torch.argmax(outputs.end_logits) + 1
return tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs["input_ids"][0][answer_start:answer_end]))
]]>
