标签:

• 医疗 许可证: mit 数据集:
• Mohammed-Altaf/medical-instruction-120k 语言:
• 英语 库名称: transformers

请注意，该聊天机器人仅设计用于研究目的，不适用于实际医疗场景。虽然该聊天机器人经过训练能提供准确有用的回答，但它不能替代专业医疗建议、诊断或治疗。聊天机器人提供的信息不应被用于做出医疗决策，任何健康问题都应咨询持证医疗专业人员。

快速开始

import torch
from transformers import GPT2LMHeadModel, GPT2Tokenizer


path = "Mohammed-Altaf/Medical-ChatBot"
device = "cuda" if torch.cuda.is_available() else "cpu"
tokenizer = GPT2Tokenizer.from_pretrained(path)
model = GPT2LMHeadModel.from_pretrained(path).to(device)

prompt_input = (
    "人类与AI助手之间的对话。\n"
    "[|人类|] {input}\n"
    "[|AI|]"
)
sentence = prompt_input.format_map({'input': "什么是帕金森病？"})
inputs = tokenizer(sentence, return_tensors="pt").to(device)

with torch.no_grad():
    beam_output = model.generate(**inputs,
                                min_new_tokens=1, 
                                max_length=512,
                                num_beams=3,
                                repetition_penalty=1.2,
                                early_stopping=True,
                                eos_token_id=198 
                                )
    print(tokenizer.decode(beam_output[0], skip_special_tokens=True))

示例输出

人类与AI助手之间的对话。
[|人类|] 什么是帕金森病？
[|AI|] 帕金森病是一种影响运动的神经退行性疾病，由大脑中产生多巴胺的细胞丧失引起。

人类与AI助手之间的对话。
[|人类|] 哪种蜂蜜对新冠咳嗽最有效？
[|AI|] 麦卢卡蜂蜜已被证明具有抗炎和抗菌特性，可帮助缓解新冠咳嗽症状。