Update app.py

app.py

@@ -99,19 +99,20 @@ def infer(image, model_selection, width, height, overlap_width, num_inference_st
         negative_pooled_prompt_embeds,
     ) = pipe.encode_prompt(final_prompt, "cpu", True)
 
-     result = pipe(
+    for image in pipe(
         prompt_embeds=prompt_embeds,
         negative_prompt_embeds=negative_prompt_embeds,
         pooled_prompt_embeds=pooled_prompt_embeds,
         negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
         image=cnet_image,
         num_inference_steps=num_inference_steps
-    ).images[0]  # 最终生成图像
+    ):
+        yield cnet_image, image
 
-    result = result.convert("RGBA")
-    cnet_image.paste(result, (0, 0), mask)
+    image = image.convert("RGBA")
+    cnet_image.paste(image, (0, 0), mask)
 
-    return background, cnet_image  # 只 return 一次，而不是 yield 两次
+    yield background, cnet_image
 
 def preload_presets(target_ratio):
     if target_ratio == "9:16":