Error al cargar el modelo

[JrGxllxgo]

Buenas, siguiendo el vídeo hice los mismos pasos pero con mi propio modelo, entrenamientos... pero me da un error, al clonar su repositorio me da el mismo error. El error salta en a la hora de cargar el modelo en un JS. No he encontrado mucha información sobre ello, podría echarme una mano? Muchas gracias!!

(async() => { console.log("Cargando modelo..."); modelo = await tf.loadLayersModel("model.json"); console.log("Modelo cargado"); })();

Uncaught (in promise) Error: Unknown layer: KerasLayer. This may be due to one of the following reasons:

1. The layer is defined in Python, in which case it needs to be ported to TensorFlow.js or your JavaScript code.
2. The custom layer is defined in JavaScript, but is not registered properly with tf.serialization.registerClass().