chore: replace shape=INPUT_SHAPE by height=256, width=256

app.py

@@ -388,7 +388,8 @@ with demo:
     )
     with gr.Row():
         input_image = gr.Image(
-            value=None, label="Upload an image here.", shape=INPUT_SHAPE, source="upload", interactive=True,
+            value=None, label="Upload an image here.",  height=256,
+            width=256, source="upload", interactive=True,
         )
 
         examples = gr.Examples(