app.py analyse function fixed

src/app.py

@@ -43,7 +43,14 @@ def load_model():
         )
         
         # Load trained weights
-        model.load_state_dict(torch.load(SAVED_MODELS_DIR / "final_model.pt", map_location=torch.device('cpu')))
+        state_dict = torch.load(SAVED_MODELS_DIR / "final_model.pt", map_location=torch.device('cpu'))
+        
+        # Filter out unexpected keys
+        model_state_dict = model.state_dict()
+        filtered_state_dict = {k: v for k, v in state_dict.items() if k in model_state_dict}
+        
+        # Load the filtered state dict
+        model.load_state_dict(filtered_state_dict, strict=False)
         model.eval()
         st.session_state.model = model