How to force the task of inference API?

Hello! I trained a model for zero-shot classification and in the pipelines it works great, but the inference api recognizes the task as text classification, and thus I can’t use it as zero-shot out of the box. Is there a way to hint the inference API to what task I want to use my model?

Thanks!

Replying to self, and figured out after looking into the code of the InferenceAPI objects. You can pass the task type as a path parameter like <<inference api base url>>/pipeline/task-type/model. Leaving it here in case anyone has the same question.

1 Like