runtime error

[===========================>..] - ETA: 0s - loss: 0.0140 - accuracy: 0.9953 1797/1875 [===========================>..] - ETA: 0s - loss: 0.0140 - accuracy: 0.9953 1809/1875 [===========================>..] - ETA: 0s - loss: 0.0141 - accuracy: 0.9952 1820/1875 [============================>.] - ETA: 0s - loss: 0.0140 - accuracy: 0.9953 1830/1875 [============================>.] - ETA: 0s - loss: 0.0140 - accuracy: 0.9953 1841/1875 [============================>.] - ETA: 0s - loss: 0.0140 - accuracy: 0.9953 1851/1875 [============================>.] - ETA: 0s - loss: 0.0140 - accuracy: 0.9953 1862/1875 [============================>.] - ETA: 0s - loss: 0.0144 - accuracy: 0.9952 1872/1875 [============================>.] - ETA: 0s - loss: 0.0143 - accuracy: 0.9952 1875/1875 [==============================] - 19s 10ms/step - loss: 0.0143 - accuracy: 0.9952 - val_loss: 0.0504 - val_accuracy: 0.9849 Traceback (most recent call last): File "/home/user/app/app.py", line 37, in <module> input_image = gr.Image(shape=(28, 28), image_mode='L', invert_colors=False) File "/usr/local/lib/python3.10/site-packages/gradio/component_meta.py", line 163, in wrapper return fn(self, **kwargs) TypeError: Image.__init__() got an unexpected keyword argument 'shape'

Container logs: