Update README.md

README.md

@@ -70,18 +70,29 @@ If applicable, mention any limitations of the model, such as inability to diagno
 ## How to Get Started with the Model
 Use the code below to get started with the model.
 
+### Read model
+
+```
+from peft import PeftModel, PeftConfig
+from transformers import AutoModelForCausalLM
+
+model_id = "zementalist/llama-3-8B-chat-psychotherapist"
+config = PeftConfig.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path)
+model = PeftModel.from_pretrained(model, model_id)
+
 ```
 
-from transformers import AutoTokenizer, AutoModel
+### Inference
+
 
-tokenizer = AutoTokenizer.from_pretrained("zementalist/llama-3-8B-chat-psychotherapist")
-model = AutoModel.from_pretrained("zementalist/llama-3-8B-chat-psychotherapist")
+```
 
-dialogue = "I feel like I don't exist and my body is not my own, I'm sombody else observing me, what causes these ideas?"
+question = "I feel like I don't exist and my body is not my own, I'm sombody else observing me, what causes these ideas?"
 
 messages = [
-    {"role": "system", "content": "Reply to the following inquiry:"},
-    {"role": "user", "content": dialogue}
+    {"role": "system", "content": "Answer the following inquiry:"},
+    {"role": "user", "content": question}
 ]
 
 input_ids = tokenizer.apply_chat_template(