Thanks for this research and this repo. I ran the zero-shot script and found some dependencies missing. Specifically, in addition to the dependencies specified in the README, I needed to install matplotlib and pandas. Perhaps update the readme or even include a requirements.txt file?