In the line activation_model = keras.Model(inputs=model.input, outputs=layer_outputs) I believe it should be `model.inputs`, as seen at https://keras.io/2/api/models/model/