Improve model card: Add pipeline tag, library name, paper link, and fix sample usage

README.md

@@ -1,8 +1,11 @@
 ---
-license: cc-by-nc-nd-4.0
 language:
 - en
+license: cc-by-nc-nd-4.0
+pipeline_tag: image-feature-extraction
+library_name: timm
 ---
+
 # Model Card for StainNet
 
 <!-- Provide a quick summary of what the model is/does. -->
@@ -11,11 +14,14 @@ language:
 
 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].
 
+This model was presented in the paper [StainNet: A Special Staining Self-Supervised Vision Transformer for Computational Pathology](https://huggingface.co/papers/2512.10326).
+
 ## Using StainNet to extract features from special staining pathology image
 
 ```python
 import timm
 import torch
+import torchvision.transforms as transforms
 
 model = timm.create_model('hf_hub:JWonderLand/StainNet', pretrained=True)