Replaced histogram-style visualizations for colored square. Removed redundant softmax column.
Replaced histogram-style visualizations for colored square. Removed redundant softmax column.

jessevig/bertviz,jessevig/bertviz