Update app.py

app.py

@@ -37,7 +37,9 @@ if 'generate' in st.session_state:
   latents = sample_latent(ROWS * COLUMNS, jax.random.PRNGKey(unique_id))
   previous = st.session_state['generate']
   if previous is not None:
-    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, value=0.25)
+    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, value=st.session_state["similarity"])
     latents = np.repeat([previous], repeats=16, axis=0) + similarity * latents
   (g_out128, _, _, _, _, _) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
   img = np.array(to_img(g_out128))