fix SDXLImageGenerator

SDXLImageGenerator.py

@@ -1,6 +1,4 @@
-import torch
-from diffusers import DiffusionPipeline
-import time
+from io import BytesIO
 
 class SDXLImageGenerator:
     def __init__(self):
@@ -21,12 +19,19 @@ class SDXLImageGenerator:
         self.pipe.to(self.device)
 
     def generate_images(self, prompts):
-        images = []
         start_time = time.time()
-        for i, prompt in enumerate(prompts):
-            gen_image = self.pipe(prompt=prompt).images[0]
-            images.append(gen_image)
+        
+        # Generate images in a batch
+        outputs = self.pipe(prompt=prompts)
+        images = outputs.images
+
+        # Convert images to PNG byte data
+        png_images = []
+        for image in images:
+            buffer = BytesIO()
+            image.save(buffer, format="PNG")
+            png_images.append(buffer.getvalue())  # PNG data in bytes
 
         end_time = time.time()
         print("Total Time SDXL: %4f seconds" % (end_time - start_time))
-        return images
+        return png_images