Update visualization notebook to use nucleus.util.vis functionality.
Update visualization notebook to use nucleus.util.vis functionality.

PiperOrigin-RevId: 314435367

google/deepvariant,google/deepvariant,google/deepvariant