Update app.py

app.py

@@ -47,7 +47,7 @@ def infer(prompt, seed, randomize_seed):
             height=int(scales[0] * 8),
             width=int(scales[0] * 8),
             generator=generator,
-        ).images
+        ).images[0]
 
     return images