Update app.py

app.py

@@ -24,7 +24,7 @@ def generate(prompt, negative_prompt):
         num_inference_steps=30,
         width=1024,
         height=1024,
-    ).images[0]
+    ).images
 
 
 gr.Interface(