Update README.md

README.md

@@ -9,6 +9,33 @@ tags:
 ---
 ![banner](banner.png)
 
+```py
+from transformers import pipeline
+
+pipe = pipeline("text-generation", model="FWKV/Fables", trust_remote_code=True)
+
+messages = [
+    {"role": "user", "content": "Write a fable about a fox and a crow."},
+]
+
+out = pipe(
+    messages,
+    max_new_tokens=300,
+    do_sample=True,
+    temperature=0.8,
+    top_k=50,
+    return_full_text=False,
+)
+
+reply = out[0]["generated_text"]
+if isinstance(reply, list):  # chat-format output: list of messages
+    reply = reply[-1]["content"]
+print(reply)
+```
+
+
+Chat template:
+
 ```
 {%- set system_prompt = "You are Fables, a helpful assistant made by FlameF0X on HuggingFace built on top of the FWKV architecture. You write stories and fables when asked." -%}
 {%- set eos = "<|endoftext|>" -%}