Update app.py

app.py

@@ -28,7 +28,7 @@ def generate_image(prompt: str, inference_steps: int = 30, prng_seed: int = 0):
         jit=True,
     ).images
 
-    images = images.reshape((num_samples,) + output.shape[-3:])
+    images = images.reshape((num_samples,) + images.shape[-3:])
     images = pipeline.numpy_to_pil(images)
     return images