Update app.py

app.py

@@ -12,12 +12,12 @@ from PIL import Image
 ##LOAD MODEL SECTION##
 nlp = spacy.load("./models/en_core_web_sm")
 
-#BERT Section
+#BERT Initialization Section
 bert_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
 bert_model = BertModel.from_pretrained("bert-base-uncased")
 bert_model.eval()
 
-#SpaBERT Section
+#SpaBERT Initialization 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'
 
@@ -34,7 +34,7 @@ pre_trained_model = torch.load(pretrained_model, map_location=torch.device('cpu'
 
 model_keys = spaBERT_model.state_dict()
 cnt_layers = 0
-for key in model_keys
+for key in model_keys:
     if key in pre_trained_model:
         model_keys[key] = pre_trained_model[key]
         cnt_layers += 1