Update README.md

README.md

@@ -75,9 +75,12 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 Use the code below to get started with the model.
 
 ```markdown
-from transformers import AutoModel
-model = AutoModel.from_pretrained("fc91/phi3-mini-instruct-full_ethics-lora")
-tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
+from transformers import AutoModelForCausalLM
+from peft import PeftModel
+
+base_model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
+peft_model_id = "fc91/phi3-mini-instruct-full_ethics-lora"
+model = PeftModel.from_pretrained(base_model, peft_model_id)
 ```
 
 ## Training Details