AIPlane2

Sleeping

PrakhAI commited on Aug 21, 2023

Commit

38e7538

1 Parent(s): c30102f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,5 +36,5 @@ num_images = st.number_input(label="Number of images to generate", min_value=1,
 if st.button('Generate Planes'):
   latents = sample_latent(num_images, jax.random.PRNGKey(int(1_000_000 * time.time())))
   (g_out128, _, _, _, _, _) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
-  img = ((np.array(gridify(g_out128)+1)*255./2.).astype(np.uint8)
   st.image(Image.fromarray(img))

 if st.button('Generate Planes'):
   latents = sample_latent(num_images, jax.random.PRNGKey(int(1_000_000 * time.time())))
   (g_out128, _, _, _, _, _) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
+  img = (np.array(gridify(g_out128)+1)*255./2.).astype(np.uint8)
   st.image(Image.fromarray(img))