Update app.py

app.py

@@ -47,4 +47,4 @@ if 'generate' in st.session_state:
         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=set_latent, args=(latents[idx],))
+          st.button(label="Generate Similar", key="%d_%d" % (unique_id, idx), on_click=set_latent, args=(latents[idx],))