Debug masked image

app.py

@@ -519,7 +519,7 @@ def generate_image(input_img, ai_gen_image, prompt):
 
     final_image = blend_details(input_img, ai_gen_image, mask_input_image)
 
-    return final_image
+    return mask_input_image, final_image
 
 def create_ui():
     """Create Gradio UI for CarViz demo"""
@@ -539,6 +539,7 @@ def create_ui():
                 # gallery = gr.Gallery(
                 #     label="Generated images", show_label=False, elem_id="gallery"
                 # )
+                masked_image = gr.Image(label="Generated Image")
                 output_image = gr.Image(label="Generated Image")
         
         # Run button
@@ -549,7 +550,7 @@ def create_ui():
                 ai_image,
                 prompt
             ], 
-            outputs=output_image
+            outputs=[masked_image, output_image]
         )     
  
     return block