Add pipeline tag and improve metadata

Hi! I'm Niels from the Hugging Face community team.

I noticed this model card is missing some metadata. I've added `pipeline_tag: image-feature-extraction` to the YAML section to make the model more discoverable for users interested in visual tokenization and feature extraction.

I've also ensured the paper, code, and project links are correctly formatted and easy to find.

README.md

@@ -1,11 +1,13 @@
 ---
+license: apache-2.0
+pipeline_tag: image-feature-extraction
 tags:
 - model_hub_mixin
 - pytorch_model_hub_mixin
-license: apache-2.0
 ---
+
 # Communication-Inspired Tokenization for Structured Image Representations
-</h1>
+
   <p align="left">
     <a href="https://araachie.github.io">Aram Davtyan</a> •
     <a href="https://www.cvg.unibe.ch/people/sahin">Yusuf Sahin</a> •
@@ -16,21 +18,23 @@ license: apache-2.0
     <a href="https://www.cvg.unibe.ch/people/favaro">Paolo Favaro</a>
   </p>
 
-Official pre-trained models for the paper: https://arxiv.org/abs/2602.20731
+Official pre-trained models for the paper: [Communication-Inspired Tokenization for Structured Image Representations](https://huggingface.co/papers/2602.20731).
 
-Project's website: https://araachie.github.io/comit/
+[Project Website](https://araachie.github.io/comit/) | [GitHub Repository](https://github.com/Araachie/comit)
   
 ## Installation
 
-Follow the instructions at https://github.com/Araachie/comit
+Follow the instructions at [https://github.com/Araachie/comit](https://github.com/Araachie/comit)
 
 ## Usage
 
 Example usage, downloading `COMiT-B` from the Hugging Face Hub:
 
 ```python
+import torch
 from comit import COMiT
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
 model = COMiT.from_pretrained('cvg-unibe/comit-b')
 model.eval().to(device)
 ```
@@ -84,7 +88,7 @@ with torch.no_grad():
 ## Licensing
 
 Unless otherwise noted, the model weights are licensed under Apache license 2.0.
-For the code licensing, see https://github.com/Araachie/comit?tab=readme-ov-file#licensing
+For the code licensing, see [https://github.com/Araachie/comit?tab=readme-ov-file#licensing](https://github.com/Araachie/comit?tab=readme-ov-file#licensing)
 
 ## Citation