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.
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 :)