Improve model card: Add tags, paper, project, and GitHub links, and citation
#1
by
nielsr HF Staff - opened
This PR enhances the model card for Yurim0507/vit-base-16-cifar10-unlearning by:
- Adding
pipeline_tag: image-classificationto the metadata for better discoverability. - Specifying
library_name: pytorchbased on the usage example. - Moving the
license: mitfrom the content to the YAML metadata. - Including relevant tags like
vit,cifar10, andmachine-unlearning. - Adding direct links to the associated paper (Unlearning Comparator), the project page (https://gnueaj.github.io/Machine-Unlearning-Comparator/), and the GitHub repository (https://github.com/gnueaj/Machine-Unlearning-Comparator/) at the top of the README for quick access.
- Adding the BibTeX citation at the end of the model card.
- Correcting a typo in the
load_modelfunction within the sample usage code (get_vit_base16tovit_base_16).
These updates make the model card more comprehensive, discoverable, and user-friendly.