search

huggingface / setfit

Efficient few-shot learning with Sentence Transformers
https://hf.co/docs/setfit
Apache License 2.0
2.25k stars 223 forks source link

Allow passing strings to model.predict #448

Closed tomaarsen closed 11 months ago

tomaarsen commented 11 months ago

Hello!

Pull Request overview

Details

As simple as checking if the input is a string, wrapping it in a list if so, and returning the first element if so.

HuggingFaceDocBuilderDev commented 11 months ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint.