Update README.md

README.md

@@ -15,12 +15,11 @@ TODO: Change Model.
 ### Usage 
 For usage, use this code block, with GPU recommended
 ```
-from transformers import AutoModelForCausalLM, AutoTokenizer
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-config = PeftConfig.from_pretrained("tmberooney/medllama")
-model = AutoModelForCausalLM.from_pretrained("daryl149/llama-2-7b-chat-hf", device_map="auto")
-model = PeftModel.from_pretrained(model, "tmberooney/medllama")
-tokenizer=AutoTokenizer.from_pretrained(config.base_model_name_or_path)
+tokenizer = AutoTokenizer.from_pretrained("tmberooney/medllama-merged")
+model = AutoModelForCausalLM.from_pretrained("tmberooney/medllama-merged")
 ```
 
 ### Intended Use