Update app.py

app.py

@@ -7,7 +7,11 @@ pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
 @spaces.GPU(duration=250)
 def generate_image(prompt):
     # Run the diffusion model to generate an image
-    image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)
+    output = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)
+    image_tensor = output.images[0]
+    image_array = image_tensor.cpu().numpy().transpose(1, 2, 0)
+    image_array = (image_array * 255).astype(np.uint8)
+    image = Image.fromarray(image_array)
     return image
 
 gr_interface = gr.Interface(