Update app.py

app.py

@@ -348,7 +348,7 @@ def run_local(base_image, base_mask, reference_image, ref_mask, seed, base_mask_
     edited_image = Image.fromarray(edited_image)
 
   
-    return image_to_base64(edited_image)
+    return [edited_image,image_to_base64(edited_image)]
 
 def update_ui(option):
     if option == "Draw Mask":
@@ -391,6 +391,7 @@ with gr.Blocks() as demo:
 
         with gr.Column(scale=1):
             baseline_gallery = gr.Gallery(label='Output', show_label=True, elem_id="gallery", height=695, columns=1)
+            base64_box = gr.Textbox(label="Base64 Output", visible=False)
             with gr.Accordion("Advanced Option", open=True):
                 seed = gr.Slider(label="Seed", minimum=-1, maximum=999999999, step=1, value=666)
                 gr.Markdown("### Guidelines")
@@ -422,6 +423,6 @@ with gr.Blocks() as demo:
 
     run_local_button.click(fn=run_local,
                             inputs=[base_image, base_mask, ref_image, ref_mask, seed, base_mask_option, ref_mask_option, text_prompt],
-                            outputs=[baseline_gallery]
+                            outputs=[baseline_gallery, base64_box]
                             )
 demo.launch()