Update app.py

app.py

@@ -184,7 +184,7 @@ def infer(image, width, height, overlap_percentage, num_inference_steps, resize_
         pooled_prompt_embeds,
         negative_pooled_prompt_embeds,
     ) = pipe.encode_prompt(final_prompt, "cuda", True)
-
+    print("hello world in infer function!!!!!!!!!!!!!!!!")
     for image in pipe(
         prompt_embeds = prompt_embeds,
         negative_prompt_embeds = negative_prompt_embeds,
@@ -198,7 +198,8 @@ def infer(image, width, height, overlap_percentage, num_inference_steps, resize_
     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."""