Returning a list of images

app.py

@@ -371,7 +371,7 @@ def generate_ai_bg(input_img, prompt):
 def generate_image(input_img, prompt):
     ai_gen_image = generate_ai_bg(input_img, prompt)
 
-    return ai_gen_image
+    return [ai_gen_image]
 
 def create_ui():
     """Create Gradio UI for CarViz demo"""