api-inference-community icon indicating copy to clipboard operation
api-inference-community copied to clipboard

Update return type for call method example pipeline for text_classification

Open davanstrien opened this issue 2 years ago • 2 comments

Currently doc string types for text_classification pipeline in docker_image/commmon have

A :obj:`list`:. The object returned should be a list of one list like [[{"label": 0.9939950108528137}]] containing:

It should be

A :obj:`list`:. The object returned should be a list of one list like [[{"label": "blah", "score": 0.9939950108528137}]] containing:

see https://github.com/huggingface/api-inference-community/pull/276#discussion_r1203001474

davanstrien avatar May 24 '23 08:05 davanstrien

Indeed. Would you like to open a PR for this?

osanseviero avatar May 24 '23 08:05 osanseviero

Indeed. Would you like to open a PR for this?

will do :)

davanstrien avatar May 24 '23 10:05 davanstrien