Update app.py

app.py

@@ -201,8 +201,7 @@ def infer(image, width, height, overlap_percentage, num_inference_steps, resize_
 
     # yield background, cnet_image
     print("#####################",type(cnet_image),"##############################")
-    return background, cnet_image
-    # return cnet_image
+    return cnet_image
 
 def clear_result():
     """Clears the result ImageSlider."""
@@ -327,8 +326,8 @@ with gr.Blocks() as demo:
                             preview_button = gr.Button("Preview alignment and mask")
 
             with gr.Column():
-                # result = gr.Image(label="Generate Image", interactive=False)
-                result = ImageSlider(label="Generated Image")
+                result = gr.Image(label="Generate Image", interactive=False)
+                # result = ImageSlider(label="Generated Image")
                 history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
                 preview_image = gr.Image(label="Preview")
 
@@ -370,12 +369,11 @@ with gr.Blocks() as demo:
                resize_option, custom_resize_percentage, prompt_input, alignment_dropdown,
                overlap_left, overlap_right, overlap_top, overlap_bottom],
         outputs=[result],
+    ).then(#Update the history gallery
+        fn=lambda x,history: update_history(x, history),
+        inputs=[result, history_gallery],
+        outputs=history_gallery,
     )
-    # ).then(#Update the history gallery
-    #     fn=lambda x,history: update_history(x, history),
-    #     inputs=[result, history_gallery],
-    #     outputs=history_gallery,
-    # )
 
     preview_button.click(
         fn=preview_image_and_mask,