Update app.py

app.py

@@ -160,12 +160,12 @@ def infer(image, width, height, overlap_width, num_inference_steps, resize_optio
         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
+    return cnet_image
 
 def clear_result():
     """Clears the result ImageSlider."""