Update app.py

app.py

@@ -235,14 +235,16 @@ Performs keypoint transform from a 7x7 gird on the source image to the target im
     with gr.Row():
         # Add an Image component to display the source image.
         image1 = gr.Image(
-            shape=(240, 240),
+            height=240,
+            width=240,
             type="pil",
             label="Source Image",
         )
 
         # Add an Image component to display the target image.
         image2 = gr.Image(
-            shape=(240, 240),
+            height=240,
+            width=240,
             type="pil",
             label="Target Image",
         )