Fixing image numpy issue

app.py

@@ -64,7 +64,7 @@ def generate(prompt, uploaded_file, prompt_addition, negative_prompt, num_images
         "num_images_per_prompt": num_images
     }
     generated_images = pipe(**params).images
-    return [numpy_to_pil(image.numpy()) for image in generated_images]
+    return generated_images  # Returning PIL images directly
 
 with gr.Blocks() as demo:
     with gr.Row():