Update app.py
Browse files
app.py
CHANGED
|
@@ -44,4 +44,5 @@ if st.button('Generate Digit'):
|
|
| 44 |
latents = sample_latent(jax.random.PRNGKey(int(1_000_000 * time.time())))
|
| 45 |
g_out = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
|
| 46 |
img = ((np.array(g_out)+1)*255./2.).astype(np.uint8)[0]
|
| 47 |
-
st.image(Image.fromarray(np.repeat(img, repeats=3, axis=2)))
|
|
|
|
|
|
| 44 |
latents = sample_latent(jax.random.PRNGKey(int(1_000_000 * time.time())))
|
| 45 |
g_out = generator.apply({'params': g_state['params'], 'batch_stats': g_state['batch_stats']}, latents, training=False)
|
| 46 |
img = ((np.array(g_out)+1)*255./2.).astype(np.uint8)[0]
|
| 47 |
+
st.image(Image.fromarray(np.repeat(img, repeats=3, axis=2)))
|
| 48 |
+
st.write("The model's details are at https://huggingface.co/PrakhAI/DigitGAN/blob/main/README.md")
|