Add pipeline tag and library name
#1
by
nielsr
HF Staff
- opened
This PR improves the model card for ufal/byt5-base-geccc-mate
by:
- Adding
pipeline_tag: text-generation
to accurately reflect its function as a grammar error correction model that generates corrected text. This ensures the model appears in the correct categories on the Hugging Face Hub (e.g., https://huggingface.co/models?pipeline_tag=text-generation). - Adding
library_name: transformers
to the metadata. This is supported by the existing sample code in the "How to Get Started with the Model" section, which usestransformers.AutoTokenizer
andtransformers.AutoModelForSeq2SeqLM
, and by the model's configuration (config.json
). This will enable the automated "Use in Transformers" widget on the model page, providing users with a convenient and ready-to-use code snippet for inference.
These changes enhance discoverability and usability for the community.
Thanks!
Solved in 49d8b99e397f60e01bfa780a8fa1c76e20595691 (the original PR also changes formatting [missing a newline at the end, among others]).
foxik
changed pull request status to
closed