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 uses transformers.AutoTokenizer and transformers.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.

Institute of Formal and Applied Linguistics, Charles University, Prague org
edited 7 days ago

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

Sign up or log in to comment