Update app.py

app.py

@@ -37,10 +37,7 @@ 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:
-    if 'similarity' not in st.session_state:
-      st.session_state['similarity'] = 0.25
-    similarity = st.number_input(label="Similarity (for \"Generate Similar\"", min_value=0.01, max_value=1.0, value=st.session_state['similarity'])
-    st.session_state['similarity'] = similarity
+    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)
     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))