edit: image generation return/

app.py

@@ -10,7 +10,6 @@ import time
 
 import gradio as gr
 import keras_cv
-from PIL import Image
 
 from constants import css, examples, img_height, img_width, num_images_to_gen
 from share_btn import community_icon_html, loading_icon_html, share_js
@@ -40,14 +39,8 @@ def generate_image_fn(prompt: str, unconditional_guidance_scale: int) -> list:
         unconditional_guidance_scale=unconditional_guidance_scale,
     )
     end_time = time.time()
-    image_paths = []
     print(f"Time taken: {end_time - start_time} seconds.")
-    for i in range(len(images)):
-        filename = f"prompt_{i}.png"
-        image = Image.fromarray(images[i])
-        image.save(filename)
-        image_paths.append(filename)
-    return image_paths
+    return images
 
 
 block = gr.Blocks(css=css)