PrakhAI commited on
Commit
f136604
·
1 Parent(s): b95d95b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -36,11 +36,9 @@ if 'generate' in st.session_state:
36
  unique_id = int(1_000_000 * time.time())
37
  latents = sample_latent(ROWS * COLUMNS, jax.random.PRNGKey(unique_id))
38
  previous = st.session_state['generate']
39
- if previous:
40
- print("Now generating similar")
41
  latents = np.repeat([previous], repeats=16, axis=0) + 0.01 * latents
42
  else:
43
- print("Now generating random")
44
  (g_out128, _, _, _, _, _) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
45
  img = np.array(to_img(g_out128))
46
  for row in range(ROWS):
 
36
  unique_id = int(1_000_000 * time.time())
37
  latents = sample_latent(ROWS * COLUMNS, jax.random.PRNGKey(unique_id))
38
  previous = st.session_state['generate']
39
+ if previous is not None:
 
40
  latents = np.repeat([previous], repeats=16, axis=0) + 0.01 * latents
41
  else:
 
42
  (g_out128, _, _, _, _, _) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
43
  img = np.array(to_img(g_out128))
44
  for row in range(ROWS):