Update app.py

app.py

@@ -193,12 +193,13 @@ def infer(image, width, height, overlap_percentage, num_inference_steps, resize_
         image = cnet_image,
         num_inference_steps=num_inference_steps
     ):
-        yield cnet_image, image
-
+         yield cnet_image, image
+        
     image = image.convert('RGBA')
     cnet_image.paste(image, (0,0), mask)
 
-    yield background, cnet_image
+    # yield background, cnet_image
+    return cnet_image
 
 def clear_result():
     """Clears the result ImageSlider."""
@@ -323,11 +324,11 @@ 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(
-                    interactive=False,
-                    label="Generated Image",
-                )
+                result = gr.Image(label="Generate Image", interactive=False)
+                # result = ImageSlider(
+                #     interactive=False,
+                #     label="Generated Image",
+                # )
                 history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
                 preview_image = gr.Image(label="Preview")