Update app.py

app.py

@@ -35,7 +35,7 @@ def generate_image(prompt: str, negative_prompt: str = "", inference_steps: int
     prompt_ids = shard(prompt_ids)
 
     if negative_prompt == "":
-        image = pipeline(
+        images = pipeline(
             prompt_ids=prompt_ids,
             params=p_params,
             prng_seed=rng,
@@ -49,7 +49,7 @@ def generate_image(prompt: str, negative_prompt: str = "", inference_steps: int
         neg_prompt_ids = pipeline.prepare_inputs(
             [negative_prompt] * num_samples)
         neg_prompt_ids = shard(neg_prompt_ids)
-        image = pipeline(
+        images = pipeline(
             prompt_ids=prompt_ids,
             params=p_params,
             prng_seed=rng,
@@ -62,7 +62,7 @@ def generate_image(prompt: str, negative_prompt: str = "", inference_steps: int
         ).images
     images = images.reshape((num_samples,) + images.shape[-3:])
     images = pipeline.numpy_to_pil(images)
-    return image[0]
+    return images[0]
 
 examples = [
     ["A watercolor painting of a bird"],