Avaliação e validação do modelo - erro de interpretação

Boa noite, durante o vídeo: " Avaliação e validação do modelo" ao tentar executar o código:

import numpy as np

teste = modelo.predict(imagens_teste) #identificador_teste
print("resultado testes: " , np.argmax(teste[0]))
print("número da imagem de teste: " , identificador_teste[np.argmax(teste[0])])

obtive o seguinte erro:

WARNING:tensorflow:Model was constructed with shape (None, 28, 28) for input KerasTensor(type_spec=TensorSpec(shape=(None, 28, 28), dtype=tf.float32, name='flatten_input'), name='flatten_input', description="created by layer 'flatten_input'"), but it was called on an input with incompatible shape (None, 10).

Podem me ajudar? Obrigado:)