Updated model load_state_dict() in FunctionsModelSA_V1.py

FunctionsModelSA_V1.py

@@ -130,7 +130,7 @@ class ToneTagger(pl.LightningModule):
 
 # LOAD IN PRE TRAINED MODEL WITH WEIGHTS
 model=ToneTagger(8) # load up the model archetecture with 8 different tones 
-model.load_state_dict(torch.load("models/SAMODEL")) # populate the weights of the model
+model.load_state_dict(torch.load("models/SAMODEL"), strict=False) # populate the weights of the model
 model.eval()