Also can you check this version of a demo I created a while back as a demo that may work slightly better. If not no worries, but worth a try.

Also @pyu May have some thoughts here as he was heavily involved in creating the model conversion on the TensorFlow.js side of things!