JasonTPhillipsJr commited on
Commit
393b919
·
verified ·
1 Parent(s): fbce538

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ bert_model = BertModel.from_pretrained("bert-base-uncased")
17
  bert_model.eval()
18
 
19
  #SpaBERT Section
20
- data_file_path = './models/spabert/datasets/SPABERT_finetuning_data_combined.json'
21
- pretrained_model = './models/spabert/datasets/fine-spabert-base-uncased-finetuned-osm-mn.pth'
22
 
23
  b_model = BertForMaskedLM.from_pretrained('bert-base-uncased')
24
  b_tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
 
17
  bert_model.eval()
18
 
19
  #SpaBERT Section
20
+ data_file_path = 'models/spabert/datasets/SPABERT_finetuning_data_combined.json'
21
+ pretrained_model = 'models/spabert/datasets/fine-spabert-base-uncased-finetuned-osm-mn.pth'
22
 
23
  b_model = BertForMaskedLM.from_pretrained('bert-base-uncased')
24
  b_tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')