Update pipeline tag, add library name and GitHub link, fix example import

README.md

@@ -1,10 +1,11 @@
 ---
-license: mit
-language:
-- en
 base_model:
 - Qwen/Qwen2.5-VL-7B-Instruct
-pipeline_tag: reinforcement-learning
+language:
+- en
+license: mit
+pipeline_tag: image-text-to-text
+library_name: transformers
 ---
 
 # 🧠 Ariadne
@@ -12,10 +13,12 @@ pipeline_tag: reinforcement-learning
 This is the official model checkpoint for the paper:  
 **[Ariadne: A Controllable Framework for Probing and Extending VLM Reasoning Boundaries](https://arxiv.org/abs/2511.00710)**  
 
+Code: https://github.com/Minghe-Shen/Ariadne
+
 ### 🔬 Example
 
 ```python
-
+import torch # Added for torch.bfloat16 and torch.cuda.is_available()
 from transformers import AutoModelForImageTextToText, AutoProcessor
 
 MODEL_ID = "..." # path
@@ -62,4 +65,5 @@ input_len = inputs["input_ids"].shape[1]
 gen_ids = sequences[0, input_len:]
 resp_text = processor.tokenizer.decode(
     gen_ids, skip_special_tokens=True, clean_up_tokenization_spaces=True
-).strip()
+).strip()
+```