#!/usr/bin/env python3

import tensorflow as tf

@tf.keras.utils.register_keras_serializable(package='Custom', name=None)
def text_standardizer(input_data):
    lowercase = tf.strings.lower(input_data)
    return lowercase

with tf.keras.utils.CustomObjectScope({'text_standardizer': text_standardizer}):
    model = tf.keras.models.load_model('final_model')
    model.summary()

    while True:
        print(model.predict([input('> ')]))