Views
No views yet
1prompt = """You are an intelligent clinical languge model.
2Below is a snippet of patient's discharge summary and a following instruction from healthcare professional.
3Write a response that appropriately completes the instruction.
4The response should provide the accurate answer to the instruction, while being concise.
5
6[Discharge Summary Begin]
7{note}
8[Discharge Summary End]
9
10[Instruction Begin]
11{question}
12[Instruction End]
13"""
14from transformers import AutoTokenizer, AutoModelForCausalLM
15tokenizer = AutoTokenizer.from_pretrained("starmpcc/Asclepius-Llama2-13B", use_fast=False)
16model = AutoModelForCausalLM.from_pretrained("starmpcc/Asclepius-Llama13-7B")
17
18note = "This is a sample note"
19question = "What is the diagnosis?"
20
21model_input = prompt.format(note=note, question=question)
22input_ids = tokenizer(model_input, return_tensors="pt").input_ids
23output = model.generate(input_ids)
24print(tokenizer.decode(output[0]))