Would be nice to see a way to save and load a model in your demo. It is a bit frustrating to deal with the tfjs lib.