If you got here, you might be searching for this:

Colab Implementation, where this specific repo is used.

https://colab.research.google.com/drive/1DKT5rFBTHhkyibVMK4SCYTJWHl2kaV3p?usp=sharing

Original implementation:

https://huggingface.co/RedRocket/JointTaggerProject

All credit goes to them.