Update app.py

app.py

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