Update README.md

README.md

@@ -2,12 +2,14 @@
 license: cc-by-nc-nd-4.0
 language:
 - en
+pipeline_tag: image-feature-extraction
+library_name: timm
 ---
 # Model Card for StainNet
 
 <!-- Provide a quick summary of what the model is/does. -->
 
-`StainNet` is a lightweight foundation model for special staining histology images.
+`StainNet` is a lightweight foundation model for special staining histology images. Arxiv preprint paper: [https://arxiv.org/abs/2512.10326]
 
 The model is a Vision Transformer Small/16 with DINO [1] self-supervised pre-training on 1,418,938 patch images from 20,231 special staining whole slide images (WSIs) in HISTAI [2].
 
@@ -16,6 +18,8 @@ The model is a Vision Transformer Small/16 with DINO [1] self-supervised pre-tra
 ```python
 import timm
 import torch
+import torchvision.transforms as transforms
+
 
 model = timm.create_model('hf_hub:JWonderLand/StainNet', pretrained=True)