Correct pipeline tag and add library name

#1
by nielsr HF staff - opened

This PR corrects the pipeline_tag from text-generation to the more appropriate feature-extraction, reflecting the model's use in information retrieval. It also adds library_name: transformers to the metadata, clarifying the model's compatibility with the Transformers library.

Center for Language and Speech Processing @ JHU org

Thanks for this Nils! I went with text generation since the model could be used in that HF “pipeline” class. I’m not entirely sure it should be feature-extraction based on these docs, but you might have more insight than me on that.

Yeah that's a different type of feature extractor :) we use pipeline_tag: feature-extraction to indicate this task: https://huggingface.co/tasks/feature-extraction.

But feel free to use text-generation if you feel that suits better :)

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment