Update app.py

app.py

@@ -285,8 +285,8 @@ with image_blocks as demo:
             # text_prompt = gr.Textbox(lines=1, label="Prompt")
             btn = gr.Button(value="Submit")
         with gr.Column():
-            image_out_seg = gr.Image(label="Segmented object", height=512, width=512)
-            image_out_shift = gr.Image(label="Shifted object", height=512, width=512)
+            image_out_seg = gr.Image(label="Segmented object", height=400, width=400)
+            image_out_shift = gr.Image(label="Shifted object", height=400, width=400)
             
     btn.click(fn=main_fun, inputs=[image, x_units, y_units, text_prompt, box_threshold, text_threshold, inpaint_text_prompt, num_inference_steps], outputs=[image_out_seg, image_out_shift])