¿Qué se debe utilizar en lugar de predict_classes() en un cuaderno Jupyter con Tensorflow? (Generación de texto NLP)

Estoy tratando de seguir un tutorial de tensorflow NLP para entrenar una red neural para generar salidas de poesía/letras-like usando mis propias fuentes compiladas. Solo conozco Python básico, así que definitivamente esto está fuera de mi nivel de competencia. Parece que el tutorial está un poco desactualizado, ya que estoy recibiendo este código de error:

AttributeError: El objeto 'Sequential' no tiene el atributo 'predict_classes'

Entiendo que el atributo ‘predict_classes’ está obsoleto y ya no se utiliza en las versiones actuales de tensorflow.

Esta fue una línea de código sugerida en una respuesta, pero no entiendo cómo incluirla en mi código:

= np.argmax(model.predict(x_test), axis=-1)

Cualquier ayuda será apreciada. Aquí está la sección del código que me está dando problemas, junto con el código de error. También incluí un enlace al cuaderno de Jupyter completo.

seed_text = "Luz de primavera"
next_words = 100

for _ in range(next_words):
    token_list = tokenizer.texts_to_sequences([seed_text])[0]
    token_list = pad_sequences([token_list], maxlen=max_sequence_len-1, padding='pre')
    predicted = model.predict_classes(token_list, verbose=0)
    output_word = ""
    for word, index in tokenizer.word_index.items():
        if index == predicted:
            output_word = word
            break
    seed_text += " " + output_word

print(seed_text)

Aquí está el enlace al video que estaba siguiendo: aquí

Y aquí está la copia del cuaderno de Jupyter: aquí