Update app.py

app.py

@@ -37,7 +37,8 @@ def generate_images(previous=None):
         with col:
           idx = row*COLUMNS + col_idx
           st.image(Image.fromarray(img[idx]))
-          st.button(label="Generate similar", key="%d_%d" % (unique_id, idx), on_click=generate_images, args=(latents[idx]))
+          if st.button(label="Generate similar", key="%d_%d" % (unique_id, idx)):
+            generate_images(latents[idx])
 
 st.write("The model and its details are at https://huggingface.co/PrakhAI/AIPlane2")
 if st.button('Generate Random'):