Update app.py

app.py

@@ -375,7 +375,11 @@ def generate_images_with_progress(prompt, negative_prompt, batch_count, use_cont
             clear_memory()  # Clear memory after each image, even in single image mode
 
     clear_memory()
-    return images
+
+    # Ensure all images are in PIL format
+    pil_images = [img if isinstance(img, Image.Image) else Image.fromarray(np.array(img)) for img in images]
+
+    return pil_images
 
 # Function to extract PNG metadata
 def extract_png_info(image_path):