This is the model that I have defined:
def build_model(): input_layer = keras.layers.Input(name="Input", shape=(MAX_LEN), dtype=\'int64\') bert