Update app.py
Browse files
app.py
CHANGED
|
@@ -57,5 +57,5 @@ if st.button('Generate Plane'):
|
|
| 57 |
latents = sample_latent(jax.random.PRNGKey(int(1_000_000 * time.time())))
|
| 58 |
(g_out32, g_out16, g_out8, g_out4) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
|
| 59 |
img = ((np.array(g_out32[0])+1)*255./2.).astype(np.uint8)
|
| 60 |
-
st.image(Image.fromarray(img)
|
| 61 |
st.write("The model's details are at https://huggingface.co/PrakhAI/AIPlane/blob/main/README.md")
|
|
|
|
| 57 |
latents = sample_latent(jax.random.PRNGKey(int(1_000_000 * time.time())))
|
| 58 |
(g_out32, g_out16, g_out8, g_out4) = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
|
| 59 |
img = ((np.array(g_out32[0])+1)*255./2.).astype(np.uint8)
|
| 60 |
+
st.image(Image.fromarray(img))
|
| 61 |
st.write("The model's details are at https://huggingface.co/PrakhAI/AIPlane/blob/main/README.md")
|