Changed output gallery to image

app.py

@@ -509,7 +509,7 @@ def blend_details(input_image, relit_image, masked_image):
         # Save final result
         final_image_pil = tensor2pil(final_image)
         # final_image_pil.save("output/output_image.png")
-        return [final_image_pil]
+        return final_image_pil
 
 def generate_image(input_img, ai_gen_image, prompt):
 
@@ -536,9 +536,10 @@ def create_ui():
                 run_button = gr.Button(value='Run')
 
             with gr.Column():
-                gallery = gr.Gallery(
-                    label="Generated images", show_label=False, elem_id="gallery"
-                )
+                # gallery = gr.Gallery(
+                #     label="Generated images", show_label=False, elem_id="gallery"
+                # )
+                output_image = gr.Image(label="Generated Image")
         
         # Run button
         run_button.click(