Update app.py

app.py

@@ -172,10 +172,8 @@ with gr.Blocks(css='style.css') as demo:
     zs = gr.State()
     do_inversion = gr.State(value=True)
     with gr.Row():
-        input_image = gr.Image(label="Input Image", interactive=True)
-        input_image.style(height=365, width=365)
-        output_image = gr.Image(label=f"Edited Image", interactive=False)
-        output_image.style(height=365, width=365)
+        input_image = gr.Image(label="Input Image", interactive=True, height=365, width=365)
+        output_image = gr.Image(label=f"Edited Image", interactive=False, height=365, width=365)
     
     with gr.Row():
         tar_prompt = gr.Textbox(lines=1, label="Describe your desired edited output", interactive=True)