Update README.md

README.md

@@ -10,14 +10,27 @@ base_model:
 <!-- markdownlint-disable no-duplicate-header -->
 
 
+
 <div align="center">
   <img src="https://github.com/abdelfattah-lab/TokenButler/blob/main/figs/tokenbutlerlogo.png?raw=true" width="50%" alt="TokenButler" />
 </div>
 <hr>
-<p align="center">
-  <a href="https://github.com/abdelfattah-lab/TokenButler/blob/main/TokenButler_Draft.pdf"><b>Paper Link</b></a>
-</p>
+<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;">
+    <img alt="Paper" 
+         src="https://img.shields.io/badge/Paper-View-orange?logo=readthedocs&logoColor=white" 
+         style="display: inline-block; vertical-align: middle;"/>
+  </a>
+  <!-- GitHub Badge -->
+  <a href="https://github.com/abdelfattah-lab/TokenButler" target="_blank" style="margin: 2px;">
+    <img alt="GitHub" 
+         src="https://img.shields.io/badge/GitHub-Repo-black?logo=github&logoColor=white" 
+         style="display: inline-block; vertical-align: middle;"/>
+  </a>
+</div>
 
+<br>
 
 
 The collection of TokenButler models can be found [here](https://huggingface.co/collections/akhauriyash/tokenbutler-67cf181b5762d0d60e5f312b). To run the `meta-llama/Llama-3.2-3B` model, follow: