Update app.py

app.py

@@ -34,9 +34,10 @@ def generate(prompt, steps=50, guidance_scale=7.5, height=768, width=512):
       width=width,
       seed=0
   )
-  image = output['images'][0]
-  image = Image.fromarray(image)
-  return image
+  return output['images']
+  # image = output['images'][0]
+  # image = Image.fromarray(image)
+  # return image
 
 with gr.Blocks(title="🔥 CHATS-SDXL Demo") as demo:
     gr.Markdown(