Update app.py

app.py

@@ -1572,9 +1572,9 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
 
                     image = self.image_processor.postprocess(image, output_type=output_type)
                     print(f"DEBUG IMAGE: {image}")
-                    image[0].save(f'{result_path}/AccDiffusion_{current_scale_num}.png')
-                    print(f"DEBUG IMAGE[Ø]: {image}")
-                    output_images.append(image[0])
+                    image.save(f'{result_path}/AccDiffusion_{current_scale_num}.png')
+                    
+                    output_images.append(image)
 
                     # cast back to fp16 if needed
                     if needs_upcasting: