Update app.py

app.py

@@ -17,11 +17,19 @@ bert_model = BertModel.from_pretrained("bert-base-uncased")
 bert_model.eval()
 
 #SpaBERT Section
+data_file_path = './models/spabert/datasets/SPABERT_finetuning_data_combined.json'
+pretrained_model = './models/spabert/datasets/fine-spabert-base-uncased-finetuned-osm-mn.pth'
 
+b_model =  BertForMaskedLM.from_pretrained('bert-base-uncased')
+b_tokenizer =  BertTokenizer.from_pretrained('bert-base-uncased')
 
+config = SpatialBertConfig()
+config.output_hidden_states = True
 
+spaBERT_model = SpatialBertForMaskedLM(config)
+spaBERT_model.load_state_dict(b_model.state_dict(), strict = False)
 
-
+pre_trained_model = torch.load(pretrained_model)