Update app.py

app.py

@@ -38,10 +38,10 @@ if 'generate' in st.session_state:
   (g_out128, _, _, _, _, _) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
   img = np.array(to_img(g_out128))
   for row in range(ROWS):
-  with st.container():
-    for (col_idx, col) in enumerate(st.columns(COLUMNS)):
-      with col:
-        idx = row*COLUMNS + col_idx
-        st.image(Image.fromarray(img[idx]))
-        if st.button(label="Generate similar", key="%d_%d" % (unique_id, idx)):
-          st.session_state['generate'] = latents[idx]
+    with st.container():
+      for (col_idx, col) in enumerate(st.columns(COLUMNS)):
+        with col:
+          idx = row*COLUMNS + col_idx
+          st.image(Image.fromarray(img[idx]))
+          if st.button(label="Generate similar", key="%d_%d" % (unique_id, idx)):
+            st.session_state['generate'] = latents[idx]