Update app.py

app.py

@@ -376,7 +376,7 @@ def generate_images_with_progress(prompt, negative_prompt, batch_count, use_cont
 
     clear_memory()
 
-    # Ensure all images are in PIL format
+    # Convert all generated images to PIL format
     pil_images = [img if isinstance(img, Image.Image) else Image.fromarray(np.array(img)) for img in images]
 
     return pil_images
@@ -557,3 +557,4 @@ if __name__ == "__main__":
     # Your Gradio interface setup here
     demo.launch(auth=("roland", "roland"), debug=True)
     clear_memory()
+