Update README.md

README.md

@@ -5,15 +5,15 @@ language:
 pipeline_tag: image-feature-extraction
 library_name: timm
 ---
-# Model Card for StainNet
+# Model Card for StainNet-Small
 
 <!-- Provide a quick summary of what the model is/does. -->
 
-`StainNet` is a lightweight foundation model for histology images from **immunohistochemistry** and **special stains**. Arxiv preprint paper: [https://arxiv.org/abs/2512.10326]
+`StainNet-Small` is a lightweight foundation model for histology images from **immunohistochemistry** and **special stains**. 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].
 
-## Using StainNet to extract features from immunohistochemistry and special staining pathology images
+## Using StainNet-Small to extract features from immunohistochemistry and special staining pathology images
 
 ```python
 import timm
@@ -40,7 +40,7 @@ with torch.no_grad():
 
 ## Citation
 
-If `StainNet` is helpful to you, please cite our work.
+If `StainNet-Small` is helpful to you, please cite our work.
 
 ```
 @misc{li2025stainnet,