Update app.py

app.py

@@ -297,13 +297,13 @@ def run_local(base_image, base_mask, reference_image, ref_mask, seed, base_mask_
     tar_image = cv2.resize(tar_image, size)
     diptych_ref_tar = np.concatenate([masked_ref_image, tar_image], axis=1)
 
-    show_diptych_ref_tar = create_highlighted_mask(diptych_ref_tar, mask_diptych)
-    show_diptych_ref_tar = Image.fromarray(show_diptych_ref_tar)
 
     tar_mask = np.stack([tar_mask,tar_mask,tar_mask],-1)
     mask_black = np.ones_like(tar_image) * 0
     mask_diptych = np.concatenate([mask_black, tar_mask], axis=1)
-
+    
+    show_diptych_ref_tar = create_highlighted_mask(diptych_ref_tar, mask_diptych)
+    show_diptych_ref_tar = Image.fromarray(show_diptych_ref_tar)
 
     diptych_ref_tar = Image.fromarray(diptych_ref_tar)
     mask_diptych[mask_diptych == 1] = 255
@@ -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=705, columns=1)
+            baseline_gallery = gr.Gallery(label='Output', show_label=True, elem_id="gallery", height=675, 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")