Joytag Colab Notebook

[Bullseye110]

Hello! I found a google colab notebook which let you run and use joy caption with gradio: https://colab.research.google.com/github/camenduru/joy-caption-jupyter/blob/main/joy_caption_jupyter.ipynb

Is there any way to use joy tag instead of joy caption on that notebook? What lines should be replaced?

I want this to be used: https://huggingface.co/spaces/fancyfeast/joytag.

Thanks!