Update README.md

README.md

@@ -84,9 +84,9 @@ The response should provide the accurate answer to the instruction, while being
 [Instruction End] 
 """
 
-from transformers import AutoTokenizer, AutoModel
-tokenizer = AutoTokenizer.from_pretrained("starmpcc/Asclepius-7B")
-model = AutoModel.from_pretrained("starmpcc/Asclepius-7B")
+from transformers import AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("starmpcc/Asclepius-7B", use_fast=False)
+model = AutoModelForCausalLM.from_pretrained("starmpcc/Asclepius-7B")
 
 note = "This is a sample note"
 question = "What is the diagnosis?"