display is ~150x slower than calculation using jupyter display and interactive ipywidgets

Question

I have a tensorflow model which I want to plot its image output for different values.

I thus made an interactive plot on collab using ipywidgets interactive, with 25