Improve model card: Add pipeline tag, library name, paper link, and fix sample usage
#1
by
nielsr
HF Staff
- opened
This PR enhances the model card for StainNet by:
- Adding
pipeline_tag: image-feature-extractionto the metadata, improving discoverability on the Hugging Face Hub under the relevant task. - Adding
library_name: timmto the metadata, as evidenced by the sample usage, which will enable the automated "how to use" widget fortimmmodels. - Including a direct link to the paper, StainNet: A Special Staining Self-Supervised Vision Transformer for Computational Pathology, in the main content for easier access.
- Correcting the sample usage code snippet by adding
import torchvision.transforms as transforms, which is necessary for the providedtransforms.Composeto function correctly.
Added.
JWonderLand
changed pull request status to
closed