I\'m trying to implement this idea by adapting from Databrick\'s Model inference using TensorFlow Keras API article. Already coded some testing piece but it hasn\'t worked y