Shrey commited on
Commit
dfbf9e3
·
1 Parent(s): 84969e5

added changes for file name

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="/Users/mv96/Downloads/vocabularies/trained_tokenizer/vocab.txt"
7
  tokenizer_1 = BertTokenizer.from_pretrained(tokenizer_path_1)
8
 
9
  #load the model path
10
- model_path="/Users/mv96/Downloads/bert_lm_10"
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