Using Zero-Shot Classification with custom masks

[Luca-Wiehe]

Hey, nice project!

I wanted to use only the second stage of this approach, i.e. zero-shot prediction with masks that I have trained myself. Is it possible to perform inference without the first part of the pipeline? If so, which file is responsible for this?

Thanks for your support :)