fix: load state dict

app.py

@@ -5,7 +5,8 @@ from model import SentimentAnalysisModel
 
 # Load the pre-trained sentiment analysis model
 model = SentimentAnalysisModel(bert_model_name="SamLowe/roberta-base-go_emotions", num_labels=7)
-model.load_state_dict(torch.load("best_model_75.pth", map_location=torch.device('cpu')))
+model.load_state_dict(torch.load("best_model_75.pth", map_location=torch.device('cpu')), strict=False)
+
 model.eval()
 
 # Mapping from predicted class to emoji