Update app.py

app.py

@@ -137,8 +137,8 @@ def main():
         with gr.Row():
             input_image = gr.Image(label="Input Image", type="pil", interactive=True)
             edited_image = gr.Image(label=f"Output Image", type="pil", interactive=False)
-            input_image.style(height=512, width=512)
-            edited_image.style(height=512, width=512)
+            input_image.scale(height=512, width=512)
+            edited_image.scale(height=512, width=512)
         
         with gr.Row(): 
             randomize_seed = gr.Radio(