Update app.py

app.py

@@ -189,8 +189,8 @@ with gr.Blocks() as demo:
     gr.Markdown("# Neural Style Transfer")
     with gr.Row():
         with gr.Column():
-            content_input = gr.Image(label="Content Image", type="numpy", image_mode="RGB", shape=(400, 400))
-            style_input = gr.Image(label="Style Image", type="numpy", image_mode="RGB", shape=(400, 400))
+            content_input = gr.Image(label="Content Image", type="numpy", image_mode="RGB", height=400, width=400)
+            style_input = gr.Image(label="Style Image", type="numpy", image_mode="RGB", height=400, width=400)
         with gr.Column():
             output_image = gr.Image(label="Output Image")