hugging2021 commited on
Commit
547ac8f
·
verified ·
1 Parent(s): a12bc1c

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -0
app.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ unmasker = pipeline('fill-mask', model='bert-base-uncased')
3
+ unmasker("Hello I'm a [MASK] model.")
4
+
5
+ from transformers import BertTokenizer, BertModel
6
+ tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
7
+ model = BertModel.from_pretrained("bert-base-uncased")
8
+ text = "Replace me by any text you'd like."
9
+ encoded_input = tokenizer(text, return_tensors='pt')
10
+ output = model(**encoded_input)
11
+
12
+
13
+ from transformers import BertTokenizer, TFBertModel
14
+ tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
15
+ model = TFBertModel.from_pretrained("bert-base-uncased")
16
+ text = "Replace me by any text you'd like."
17
+ encoded_input = tokenizer(text, return_tensors='tf')
18
+ output = model(encoded_input)