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 == "":
-        images = pipeline(
+        image = 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)
-        images = pipeline(
+        image = pipeline(
             prompt_ids=prompt_ids,
             params=p_params,
             prng_seed=rng,
@@ -60,10 +60,7 @@ def generate_image(prompt: str, negative_prompt: str = "", inference_steps: int
             guidance_scale=float(guidance_scale),
             jit=True,
         ).images
-    images_f = []
-    image_b64 = (f"data:image/jpeg;base64,{images[0]}")
-    images_f.append(image_b64)
-    return images_f
+    return image
 
 examples = [
     ["A watercolor painting of a bird"],