Update app.py

app.py

@@ -43,7 +43,7 @@ def infer(prompt):
                 num_inference_steps=steps,
                 guidance_scale=scale,
                 #generator=generator,
-            ).images
+            ).images[0]
 
     return images