Add an Input
layer to define the input shape explicitly:
model = Sequential([
Input(shape=(28, 28)),
Flatten(),
Dense(128, activation='relu'),
Dense(10, activation='softmax')
])
Reconvert your model and try loading it again—it should work!