Neuron view

Hi,

great work. I have tried the neuron view but this is the results and also in the Colab I see like this.

For what I understood, it should be possible to see something like:

Like I see in this tutorial: https://towardsdatascience.com/deconstructing-bert-part-2-visualizing-the-inner-workings-of-attention-60a16d86b5c1

Thank you very much