Spaces:
Running
Running
Update app.py
Browse files
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 = '
|
21 |
-
pretrained_model = '
|
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')
|