Solved Example Error issue

app.py

@@ -4,6 +4,9 @@ import random
 from diffusers import AuraFlowPipeline
 import torch
 import spaces
+import uuid
+import random
+import os
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -28,37 +31,45 @@ 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",
+    "An alien grasping a sign board contain word 'AuraFlow', futuristic, neonpunk, detailed",
+    "a cute robot artist painting on an easel, concept art, light colors, cute",
+    'A vibrant street wall covered in colorful graffiti, the centerpiece spells "FIRE", in a storm of colors',
+    'Digital art, portrait of an anthropomorphic roaring Tiger warrior with full armor, close up in the middle of a battle, behind him there is a banner with the text "Open Source"'
+    "Astronaut in a jungle grasping a sign board contain word 'I love SPACE', cold color palette, muted colors, detailed, futuristic",
     "A delicious ceviche cheesecake slice",
+    "a ROBOT riding a BLUE horse on Mars, photorealistic",
 ]
 
+    
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 520px;
+    max-width: 600px;
 }
 """
 
@@ -88,13 +99,13 @@ 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,
+                lines=1,
                 placeholder="Enter a negative prompt",
             )
             
@@ -149,7 +160,7 @@ with gr.Blocks(css=css) as demo:
             fn = infer,
             inputs = [prompt],
             outputs = [result, seed],
-            cache_examples="lazy"
+            cache_examples=True
         )
 
     gr.on(