Update to version 1.4.1

Author: jessevig

README.md

@@ -16,8 +16,6 @@ BertViz is an interactive tool for visualizing attention in [Transformer](https:
    [Tensor2Tensor visualization tool](https://github.com/tensorflow/tensor2tensor/tree/master/tensor2tensor/visualization)
     by [Llion Jones](https://medium.com/@llionj), providing multiple views that each offer a unique lens into the attention mechanism.
 
-Get updates for this and related projects on [Twitter
-![Twitter logo](images/twitter.svg)](https://twitter.com/jesse_vig).
 
 ## 🚀 Quick Tour
 
@@ -67,7 +65,7 @@ pip install ipywidgets
 To create a new Jupyter notebook, simply run:
 
 ```bash
-jupyter notebook
+jupyter lab
 ```
 
 Then click `New` and select `Python 3 (ipykernel)` if prompted.
@@ -111,7 +109,7 @@ You may also run any of the sample [notebooks](notebooks/) included with BertViz
 ```bash
 git clone --depth 1 [email protected]:jessevig/bertviz.git
 cd bertviz/notebooks
-jupyter notebook
+jupyter lab
 ```
 ## 🕹 Interactive Tutorial
 

bertviz/head_view.py

@@ -2,7 +2,7 @@
 import os
 import uuid
 
-from IPython.core.display import display, HTML, Javascript
+from IPython.display import display, HTML, Javascript
 
 from .util import format_special_chars, format_attention, num_layers
 

bertviz/model_view.py

@@ -2,7 +2,7 @@
 import os
 import uuid
 
-from IPython.core.display import display, HTML, Javascript
+from IPython.display import display, HTML, Javascript
 
 from .util import format_special_chars, format_attention, num_layers, num_heads
 

bertviz/neuron_view.py

@@ -31,7 +31,7 @@
 from collections import defaultdict
 
 import torch
-from IPython.core.display import display, HTML, Javascript
+from IPython.display import display, HTML, Javascript
 
 
 def show(model, model_type, tokenizer, sentence_a, sentence_b=None, display_mode='dark', layer=None, head=None,

setup.py

@@ -10,7 +10,7 @@
 # This call to setup() does all the work
 setup(
     name="bertviz",
-    version="1.4.0",
+    version="1.4.1",
     description="Attention visualization tool for NLP Transformer models.",
     long_description=README,
     long_description_content_type="text/markdown",
@@ -19,5 +19,5 @@
     license="Apache 2.0",
     packages=["bertviz"],
     include_package_data=True,
-    install_requires=["transformers>=2.0", "torch>=1.0", "tqdm", "boto3", "requests", "regex", "sentencepiece"],
+    install_requires=["transformers>=2.0", "torch>=1.0", "tqdm", "boto3", "requests", "regex", "sentencepiece", "IPython>=7.14"],
 )