Update app.py

app.py

@@ -3,6 +3,7 @@ from diffusers import AutoPipelineForText2Image
 from io import BytesIO
 from generate_propmts import generate_prompt
 from concurrent.futures import ThreadPoolExecutor, as_completed
+import PIL
 
 # Load the model once outside of the function
 model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
@@ -14,11 +15,7 @@ def generate_image(prompt):
 
         # Check if the model returned images
         if isinstance(output.images, list) and len(output.images) > 0:
-            image = output.images[0]
-            buffered = BytesIO()
-            image.save(buffered, format="JPEG")
-            image_bytes = buffered.getvalue()
-            return image_bytes
+            return output.images[0]
         else:
             raise Exception("No images returned by the model.")