Add support for returning word attributions & visualization
Update function tag_text
to allow returning word attributions for the prediction(s). NB! It should be optional to output attributions as calculating them makes the function significantly slower.
+ Add support for visualizing the results in Jupyter Notebook (should also be configurable via a parameter as it can be only used in Jupyter).