PrakhAI commited on
Commit
0f161cf
·
1 Parent(s): 56615fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ if 'generate' in st.session_state:
34
  latents = sample_latent(ROWS * COLUMNS, jax.random.PRNGKey(unique_id))
35
  previous = st.session_state['generate']
36
  if previous:
37
- latents = np.repeat([previous], repeats=16, axis=0) + 0.25 * latents
38
  (g_out128, _, _, _, _, _) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
39
  img = np.array(to_img(g_out128))
40
  for row in range(ROWS):
 
34
  latents = sample_latent(ROWS * COLUMNS, jax.random.PRNGKey(unique_id))
35
  previous = st.session_state['generate']
36
  if previous:
37
+ latents = np.repeat([previous], repeats=16, axis=0) + 0.01 * latents
38
  (g_out128, _, _, _, _, _) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
39
  img = np.array(to_img(g_out128))
40
  for row in range(ROWS):