Spaces:
Sleeping
Sleeping
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')
|