Upload dc.py

dc.py

@@ -726,7 +726,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         True, None, None, "plus_face", "original", 0.7, None, None, "base", "style", 0.7, 0.0,
         load_lora_cpu, verbose_info, gpu_duration
     ):
-        images = stream_images
+        images = stream_images if isinstance(stream_images, list) else images
     progress(1, desc="Inference completed.")
     print(images)#
     output_image = images[0][0] if images else None