Update README.md

README.md

@@ -8,6 +8,10 @@ tags:
 - ehr
 - reasoning
 - qwen
+language:
+- en
+base_model:
+- Qwen/Qwen3-1.7B
 ---
 
 # EHR-R1-1.7B: A Reasoning-Enhanced Foundational Language Model for Electronic Health Record Analysis
@@ -34,14 +38,14 @@ This repository contains the **EHR-R1-1.7B** model, part of the **EHR-R1** serie
 ### EHR Input Format
 For any EHR data, keep the EHR input with markdown format as below:
 *   For the event with single record:
-```markdown
+```
 ## Evant Name [Event Time (YYYY-MM-DD HH:MM:SS)]
 - ItemKey_1: ItemValue_1
 - ItemKey_2: ItemValue_2
 - ItemKey_3: ItemValue_3
 ```
 *   For the event with multiple records (like labevents):
-```markdown
+```
 ## Evant Name [Event Time (YYYY-MM-DD HH:MM:SS)]
  |  ItemKey_1  |  ItemKey_2  |  ItemKey_3  |
  |  ---------  |  ---------  |  ---------  |
@@ -67,8 +71,7 @@ ehr_input = "{YOUR FOMATTED EHR INPUT}"
 instruction = "{YOUR TASK INSTRUCTION}"
 messages = [
     {"role": "system", "content": "You are a helpful assistant."},
-    {"role": "user", "content": ehr_input + "
-" + instruction}
+    {"role": "user", "content": ehr_input + "\n" + instruction}
 ]
 
 # For EHR-R1-1.7B & EHR-R1-8B, control the reasoning mode by setting enable_thinking
@@ -78,14 +81,8 @@ text = tokenizer.apply_chat_template(
     add_generation_prompt=True,
     enable_thinking=False,
 ).to(model.device)
-# For EHR-R1-72B, you can manually add <think>
-
-</think>
- at the end of the model_inputs to close the reasoning modes.
-text += "<think>
-
-</think>
-"
+# For EHR-R1-72B, you can manually add `<think>\n\n</think>` at the end of the model_inputs to close the reasoning modes.
+text += "<think>\n\n</think>"
 
 model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
 generated_ids = model.generate(