Create app.py

app.py

@@ -0,0 +1,7 @@
+from transformers import AutoTokenizer
+from bertviz.transformers_neuron_view import BertModel from bertviz.neuron_view import show
+    model_ckpt = "bert-base-uncased"
+    tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
+    model = BertModel.from_pretrained(model_ckpt)
+    text = "time flies like an arrow"
+    show(model, "bert", tokenizer, text, display_mode="light", layer=0, head=8)