Update app.py

app.py

@@ -337,7 +337,7 @@ def run_local(base_image, base_mask, reference_image, ref_mask, seed, base_mask_
     edited_image = Image.fromarray(edited_image)
 
     if ref_mask_option != "Label to Mask":
-        return [edited_image, show_diptych_ref_tar]
+        return [show_diptych_ref_tar, edited_image]
     else:
         return [return_ref_mask, show_diptych_ref_tar, edited_image]   
 
@@ -381,7 +381,7 @@ with gr.Blocks() as demo:
                 text_prompt = gr.Textbox(label="Label", placeholder="Enter the category of the reference object, e.g., car, dress, toy, etc.")
 
         with gr.Column(scale=1):
-            baseline_gallery = gr.Gallery(label='Output', show_label=True, elem_id="gallery", height=675, columns=1)
+            baseline_gallery = gr.Gallery(label='Output', show_label=True, elem_id="gallery", height=690, columns=1)
             with gr.Accordion("Advanced Option", open=True):
                 seed = gr.Slider(label="Seed", minimum=-1, maximum=999999999, step=1, value=666)
                 gr.Markdown("### Guidelines")