Update app.py

app.py

@@ -29,7 +29,7 @@ def predict(images, eps):
     grid_zca = make_grid(zca_images, nrow=3, normalize=True).cpu().numpy()
     return np.transpose(grid_zca, [1, 2, 0])
 
-def load_example_images(example_images):
+def load_example_images():
     return example_images
 
 title = 'ZCA Whitening with Kornia!'
@@ -56,7 +56,7 @@ with gr.Blocks(title=title) as demo:
     gr.Markdown("## Example Images")
     example_gallery = gr.Gallery(value=example_images, label="Example Images", columns=5, height="auto")
     load_examples_button = gr.Button("Load Example Images")
-    load_examples_button.click(fn=load_example_images, inputs=[example_gallery], outputs=[input_images])
+    load_examples_button.click(fn=load_example_images, inputs=[], outputs=[input_images])
 
 if __name__ == "__main__":
     demo.launch(show_error=True)