TrungNQ commited on
Commit
d9a904e
·
verified ·
1 Parent(s): 5922725

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,8 +1,8 @@
1
  from transformers import AutoTokenizer
2
  from bertviz.transformers_neuron_view import BertModel
3
  from bertviz.neuron_view import show
4
- model_ckpt = "bert-base-uncased"
5
- tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
6
- model = BertModel.from_pretrained(model_ckpt)
7
- text = "time flies like an arrow"
8
- show(model, "bert", tokenizer, text, display_mode="light", layer=0, head=8)
 
1
  from transformers import AutoTokenizer
2
  from bertviz.transformers_neuron_view import BertModel
3
  from bertviz.neuron_view import show
4
+ model_ckpt = "bert-base-uncased"
5
+ tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
6
+ model = BertModel.from_pretrained(model_ckpt)
7
+ text = "time flies like an arrow"
8
+ show(model, "bert", tokenizer, text, display_mode="light", layer=0, head=8)