Update app.py

app.py

@@ -40,7 +40,7 @@ if 'generate' in st.session_state:
     if "similarity" not in st.session_state:
       st.session_state["similarity"] = 0.5
     similarity = st.number_input(label="Mutation (for \"Generate Similar\") - lower value generates more similar images", key="similarity", min_value=0.01, max_value=1.0)
-    latents = np.repeat([previous], repeats=4, axis=0) + similarity * latents
+    latents = np.repeat([previous], repeats=ROWS*COLUMNS, axis=0) + similarity * latents
   g_out = generator.apply({'params': g_state['params']}, latents)
   img = np.array(to_img(g_out))
   for row in range(ROWS):