Update README.md

README.md

@@ -13,9 +13,24 @@ license: apache-2.0
 + Completed 2 epochs
 + 2e-5 learning rate
 
-## Evaluations 
+## Code Example
+
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model_id = "macadeliccc/WestLake-7B-v2-laser-truthy-dpo"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+
+model = AutoModelForCausalLM.from_pretrained(model_id)
 
-This model is experimental and this finetune may or may not retain its original intentions.
+text = "Come up with an idea for a TV show and write a short pilot script"
+inputs = tokenizer(text, return_tensors="pt")
+
+outputs = model.generate(**inputs, max_new_tokens=4096)
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
+
+## Evaluations 
 
 <pre>----Benchmark Complete----
 2024-01-27 16:44:07