Mi modelo se define como tal:

model = keras.models.Sequential()
model.add(layers.Embedding(max_features, 128, input_length=max_len,
                       input_shape=(max_len,), name='embed'))
model.add(layers.Conv1D(32, 7, activation='relu'))
model.add(layers.MaxPooling1D(5))
model.add(layers.Conv1D(32, 7, activation='relu'))
model.add(layers.GlobalMaxPooling1D())
model.add(layers.Dense(1))

y cuando uso la función plot_model para dibujarlo:

from keras.utils import plot_model

plot_model(model, show_shapes=True, to_file='model.png')

El dibujo que obtengo es

Donde la capa de entrada es una serie de números. ¿Alguien sabe cómo le permite mostrar la entrada correctamente?