Update README.md

README.md

@@ -115,10 +115,12 @@ of the results generated.
 Use the code below to get started with the model.
 
 ```
-HUB_MODEL_ID = '/somosnlp/spanish_medica_llm'
-config = PeftConfig.from_pretrained(HUB_MODEL_ID)
-model = AutoModelForCausalLM.from_pretrained(HUB_MODEL_ID, return_dict=True, load_in_8bit=True, device_map={"":0})
+from peft import PeftModel, PeftConfig
+from transformers import AutoModelForCausalLM
 
+config = PeftConfig.from_pretrained("somosnlp/spanish_medica_llm")
+model = AutoModelForCausalLM.from_pretrained("BioMistral/BioMistral-7B")
+model = PeftModel.from_pretrained(model, "somosnlp/spanish_medica_llm")
 ```
 
 ## Training Details