Update app.py

app.py

@@ -49,4 +49,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],))