Custom components: restore Tensorflow model in load function of the Component class

Hi all

I’m building a new custom component with Tensorflow.I have managed to save my model with the ‘predict’ function but I’m not able to restore it (I want to do something like this Save and Restore  |  TensorFlow). I guess that I should do it with the ‘load’ function. I’ve seen the documentation in Custom Components and several examples in the source code, but I don’t understand how this ‘load’ function works. Any suggestions, documentation or simple examples?

Thanks in advance!