Update app.py

app.py

@@ -1495,7 +1495,7 @@ def handle_generation(h,w,d):
             num_inference_steps=image_steps,
             max_sequence_length=seq,
             generator=torch.Generator(device).manual_seed(random.randint(0, MAX_SEED))
-        )
+        ).images[0]
 
         working = False