Update app.py
Browse files
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):
|