Hi,
Thanks @nncv for reporting the issue and thanks @lgusm for looking it  .
The method tfdf.model_plotter.plot_model is expected to be independent of colab, but there is probably an issue. I’ll take a look and comeback to you.
ETA: Tomorow.