Can you describe what you mean by "it doesn't seem to recognize the updated trained model", please?
The only exceptions to this that would prevent the Scratch project from loading would be if:
1) You delete your project from the Machine Learning for Kids website
(if you delete the project, all of the information about your machine learning project is lost - and that is needed to enable the custom machine learning block in the Scratch project)
2) You stored your machine learning project in the web browser (not the cloud) and then tried to load the project using a different web browser
(If you store your machine learning project in the cloud, it can be used from any computer. If you store it on your own computer, it can only be used from that computer)
Kind regards
D