Неправильное распознавание изображений в Tensorflow js

Есть код свёрточной нейронной сети в Google Colab. Нейронка обучена распознавать два класса изображений, точность на тестовых данных 85%. В колабе всё хорошо распознаётся, однако, при переносе обученной модели в tensorflow js изображения перестают правильно распознаваться, в чём может быть проблема?

Ссылка на колаб: https://colab.research.google.com/drive/17iiZTEGn61l47aEAeRbp_jpU8FN2zhqX?usp=sharing

За основу взял простое приложение с гитхаба: https://github.com/dkreider/tensorflowjs-cat-vs-dog

В данном приложении заменил обученную модель на свою, но ожидаемого результата не достиг.

Код, используемый для конвертации модели keras в модель tensorflow js:

import tensorflowjs as tfjs
tfjs.converters.save_keras_model(model, "trained-model")