Update app.py

app.py

@@ -23,7 +23,7 @@ bert_model.eval()
 #SpaBERT Initialization Section
 data_file_path = 'models/spabert/datasets/SPABERT_finetuning_data_combined.json'
 pretrained_model_path = 'models/spabert/datasets/fine-spabert-base-uncased-finetuned-osm-mn.pth'
-pre_trained_model = torch.load(pretrained_model_path, map_location=torch.device('cpu'))
+
 
 #b_model =  BertForMaskedLM.from_pretrained('bert-base-uncased')
 #b_tokenizer =  BertTokenizer.from_pretrained('bert-base-uncased')
@@ -32,6 +32,7 @@ config = SpatialBertConfig()
 config.output_hidden_states = True
 spaBERT_model = SpatialBertForMaskedLM(config)
 
+pre_trained_model = torch.load(pretrained_model_path, map_location=torch.device('cpu'))
 spaBERT_model.load_state_dict(bert_model.state_dict(), strict = False)
 spaBERT_model.load_state_dict(pre_trained_model, strict=False)