Views
No views yet

1import tensorflow as tf
2import matplotlib.pyplot as plt
3from huggingface_hub import from_pretrained_keras
4
5seed = 42
6n_images = 36
7codings_size = 100
8generator = from_pretrained_keras("huggan/crypto-gan")
9
10def generate(generator, seed):
11 noise = tf.random.normal(shape=[n_images, codings_size], seed=seed)
12 generated_images = generator(noise, training=False)
13
14 fig = plt.figure(figsize=(10, 10))
15 for i in range(generated_images.shape[0]):
16 plt.subplot(6, 6, i+1)
17 plt.imshow(generated_images[i, :, :, :])
18 plt.axis('off')
19 plt.savefig("samples.png")
20
21generate(generator, seed)