added changes for file name
Browse files
app.py
CHANGED
@@ -3,11 +3,11 @@ from transformers import FillMaskPipeline ,DistilBertTokenizer,TFAutoModelForMas
|
|
3 |
from transformers import BertTokenizer
|
4 |
|
5 |
#load the tokenizer
|
6 |
-
tokenizer_path_1="
|
7 |
tokenizer_1 = BertTokenizer.from_pretrained(tokenizer_path_1)
|
8 |
|
9 |
#load the model path
|
10 |
-
model_path="
|
11 |
model_1 = TFAutoModelForMaskedLM.from_pretrained(model_path)
|
12 |
|
13 |
#build the unmasker pipeline using HF for inference
|
|
|
3 |
from transformers import BertTokenizer
|
4 |
|
5 |
#load the tokenizer
|
6 |
+
tokenizer_path_1="./vocab.txt"
|
7 |
tokenizer_1 = BertTokenizer.from_pretrained(tokenizer_path_1)
|
8 |
|
9 |
#load the model path
|
10 |
+
model_path="./bert_lm_10"
|
11 |
model_1 = TFAutoModelForMaskedLM.from_pretrained(model_path)
|
12 |
|
13 |
#build the unmasker pipeline using HF for inference
|