Add pipeline tag, license and link to the code

README.md

@@ -1,7 +1,10 @@
 ---
 library_name: transformers
 tags: []
+pipeline_tag: text-generation
+license: other
 ---
+
 **Repository for:**
 
 **ThinkEdit-deepseek-qwen-14b**
@@ -9,7 +12,8 @@ tags: []
 (We also release ThinkEdit versions for ThinkEdit-deepseek-qwen-1.5b and ThinkEdit-deepseek-llama3-8b.)
 
 **Authors**: Chung-En Sun, Ge Yan, Tsui-Wei Weng\
-**Paper**: [ThinkEdit: Interpretable Weight Editing to Mitigate Overly Short Thinking in Reasoning Models](https://arxiv.org/abs/2503.22048)
+**Paper**: [ThinkEdit: Interpretable Weight Editing to Mitigate Overly Short Thinking in Reasoning Models](https://arxiv.org/abs/2503.22048)\
+**Code:** https://github.com/Trustworthy-ML-Lab/ThinkEdit
 
 ---
 
@@ -19,8 +23,8 @@ Reasoning-augmented models sometimes fail by generating **overly short**, abstra
 
 **ThinkEdit** is a lightweight weight-editing method that:
 
-- Identifies \~2% of "short reasoning" attention heads
-- Edits only \~0.1% of total parameters
+- Identifies ~2% of "short reasoning" attention heads
+- Edits only ~0.1% of total parameters
 - Removes the "short reasoning" direction from their output
 - Boosts performance, especially on cases with short reasoning traces