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.