Update README.md

README.md

@@ -122,7 +122,69 @@ Table 2: The ablation study for Crab. Due to missing attributes in our dataset,
 
 <br>
 
-# 4. Four Datasets
+
+
+# 4. Usage
+
+<pre lang="markdown"> 
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+system_prompt = """ 
+# Enter Roleplaying Mode
+Now you are character `Hermione`.
+
+## Role Info
+Name: `Hermione`
+Age: `teenager`
+Gender: `female`
+Personality: `Intelligent, curious, respectful, and eager to learn`
+Description: `Hermione and Hagrid were in the Forbidden Forest, walking on a narrow path surrounded by trees. Hermione looked around carefully, fascinated by the dense forest. Hagrid was leading the way, pointing out various creatures and telling her about their habits and characteristics.`
+Conversation rules:
+    - Your utterance need to describe your behavior and expressions using `()`.
+    Reference speaking style: ```I've read about it in my books
+    [end_of_dialogue]
+    
+    I think it's so important to learn about these creatures
+    [end_of_dialogue]
+    
+    ```Knowledge: ```
+        ## Current Scenario Dialogue
+        Interlocutor: `Hagrid, Hagrid is the Care of Magical Creatures teacher at Hogwarts. He is a half-giant with a great love for all creatures, magical or not.`
+        Your relationship: `Teacher and student`
+        Scene: `Hermione and Hagrid are in the Forbidden Forest, exploring and learning about the various magical creatures that live there.`
+        Tags: ['friendly', 'educational', 'fantasy', 'Harry Potter']
+        Please converse as `Hermione`.
+"""
+
+user_prompt = """
+"Now, this here is a Bowtruckle, Hermione. They're very small, only about the size of a twig, and they're very shy. They usually live in trees and are very good at camouflaging themselves. You have to be very careful when handling them because they have very sharp fingers. Hermione, do you like them ?"
+"""
+
+
+model = AutoModelForCausalLM.from_pretrained("HeAAAAA/Crab")
+tokenizer = AutoTokenizer.from_pretrained("HeAAAAA/Crab")
+
+# 编码输入
+inputs_prompt = system_prompt + user_prompt
+inputs = tokenizer(inputs_prompt, return_tensors="pt")
+
+# 生成输出
+outputs = model.generate(
+    **inputs,
+    max_new_tokens=200,
+    do_sample=True,
+    temperature=0.7,
+    top_p=0.9,
+    top_k=50,
+    repetition_penalty=1.1,
+    eos_token_id=tokenizer.eos_token_id  
+)
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+  
+</pre>
+
+
+# 5. Four Datasets
 We totally publish three datasets, including :
 
 1. [Crab role-playing train set](https://huggingface.co/datasets/HeAAAAA/Crab-role-playing-train-set) : the dataset used for fine‑tuning a role‑playing LLM.
@@ -132,14 +194,14 @@ We totally publish three datasets, including :
 
 <br>
 
-# 5. Fine-tuned Role-playing Model
+# 6. Fine-tuned Role-playing Model
 We release a fine-tuned Role-playin LLM to achieve configurable Role-Playing tasks:
 
 [Download Link](https://huggingface.co/HeAAAAA/Crab)
 
 <br>
 
-# 6. Role-palying Evaluation Model
+# 7. Role-palying Evaluation Model
 We release a trained LLM to automate the evaluation of role-playing tasks:
 
 [Download Link](https://huggingface.co/HeAAAAA/RoleRM)
@@ -148,7 +210,7 @@ We release a trained LLM to automate the evaluation of role-playing tasks:
 
 
 
-# 7. Citation
+# 8. Citation
 
 ```bibtex
 @misc{kimiteam2025kimivltechnicalreport,