Update app.py

app.py

@@ -32,7 +32,19 @@ spaBERT_model.load_state_dict(b_model.state_dict(), strict = False)
 
 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
+    if key in pre_trained_model:
+        model_keys[key] = pre_trained_model[key]
+        cnt_layers += 1
+    else:
+        print("No weight for", key)
+print(cnt_layers, 'layers loaded')
 
+spaBERT_model.load_state_dict(model_keys)
+spaBERT_model.to(device)
+spaBERT_model.eval()