Solved Example Error issue

app.py

@@ -4,6 +4,8 @@ import random
 from diffusers import AuraFlowPipeline
 import torch
 import spaces
+import uuid
+import os
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -28,37 +30,46 @@ pipe = AuraFlowPipeline.from_pretrained(
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
+def save_image(img):
+    unique_name = str(uuid.uuid4()) + ".png"
+    img.save(unique_name)
+    return unique_name
+
 @spaces.GPU
-def infer(prompt, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=5.0, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=5.0, num_inference_steps=30, progress=gr.Progress(track_tqdm=True)):
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
         
     generator = torch.Generator().manual_seed(seed)
     
-    image = pipe(
-        prompt = prompt, 
-        negative_prompt = negative_prompt,
-        width=width,
-        height=height,
-        guidance_scale = guidance_scale, 
-        num_inference_steps = num_inference_steps, 
-        generator = generator
-    ).images[0] 
-    
-    return image, seed
+    options = { "prompt" : prompt, 
+        "negative_prompt" : negative_prompt,
+        "width":width,
+        "height":height,
+        "guidance_scale" : guidance_scale, 
+        "num_inference_steps" : num_inference_steps, 
+        "generator" : generator }
+    images = pipe(**options).images
+    image_paths = [save_image(img) for img in images]
+    return image_paths, seed
 
 examples = [
     "A photo of a lavender cat",
-    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
-    "An astronaut riding a green horse",
-    "A delicious ceviche cheesecake slice",
+    "Astronaut in a jungle grasping a sign board contain word 'I love SPACE', cold color palette, muted colors, detailed, 8k",
+    "a cat eating a piece of cheese",
+    "a ROBOT riding a BLUE horse on Mars, photorealistic",
+    "a cute robot artist painting on an easel, concept art",
+    "An alien grasping a sign board contain word 'AuraFlow', futuristic, neonpunk, detailed",
+    "Kids going to school, sketch",
+    "Logo of Company name SpaceY"
 ]
 
+    
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 520px;
+    max-width: 600px;
 }
 """
 
@@ -88,14 +99,16 @@ with gr.Blocks(css=css) as demo:
             
             run_button = gr.Button("Run", scale=0)
         
-        result = gr.Image(label="Result", show_label=False)
+        result = gr.Gallery(label="Result", columns=1, show_label=False)
 
         with gr.Accordion("Advanced Settings", open=False):
             
             negative_prompt = gr.Text(
                 label="Negative prompt",
-                max_lines=1,
+                max_lines=5,
+                lines=4,
                 placeholder="Enter a negative prompt",
+                value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW",
             )
             
             seed = gr.Slider(
@@ -149,7 +162,7 @@ with gr.Blocks(css=css) as demo:
             fn = infer,
             inputs = [prompt],
             outputs = [result, seed],
-            cache_examples="lazy"
+            cache_examples=True
         )
 
     gr.on(