Add text-generation pipeline tag

README.md

@@ -1,9 +1,11 @@
 ---
-license: mit
-library_name: transformers
 base_model:
 - meta-llama/Llama-3.2-1B
+library_name: transformers
+license: mit
+pipeline_tag: text-generation
 ---
+
 # TokenButler
 <!-- markdownlint-disable first-line-h1 -->
 <!-- markdownlint-disable html -->
@@ -17,7 +19,7 @@ base_model:
 <hr>
 <div align="center" style="line-height: 1;">
   <!-- Paper Badge -->
-  <a href="https://github.com/abdelfattah-lab/TokenButler/blob/main/TokenButler_Draft.pdf" target="_blank" style="margin: 2px;">
+  <a href="https://arxiv.org/abs/2503.07518" target="_blank" style="margin: 2px;">
     <img alt="Paper" 
          src="https://img.shields.io/badge/Paper-View-orange?logo=readthedocs&logoColor=white" 
          style="display: inline-block; vertical-align: middle;"/>