Dimitre commited on
Commit
f8a7901
·
1 Parent(s): 7ac0603

Update app.py

Browse files

CryptoGAN first iteration

Files changed (1) hide show
  1. app.py +27 -4
app.py CHANGED
@@ -1,7 +1,30 @@
1
  import gradio as gr
 
 
 
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- iface.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ import tensorflow as tf
3
+ import matplotlib.pyplot as plt
4
+ from huggingface_hub import from_pretrained_keras
5
 
 
 
6
 
7
+ n_images = 64
8
+ codings_size = 100
9
+ hub_generator = from_pretrained_keras("huggan/crypto-gan")
10
+
11
+ def post_process(image):
12
+ image = ((image + 1) / 2) # scale back
13
+ return image
14
+
15
+ def generate():
16
+ noise = tf.random.normal([n_images, codings_size])
17
+ generated_images = hub_generator(noise, training=False)
18
+
19
+ fig = plt.figure(figsize=(10, 10))
20
+ for i in range(generated_images.shape[0]):
21
+ plt.subplot(8, 8, i+1)
22
+ plt.imshow(post_process(generated_images[i, :, :, :]), cmap="binary")
23
+ plt.axis('off')
24
+ return fig
25
+
26
+ gr.Interface(fn=generate,
27
+ inputs=[number_of_examples],
28
+ outputs="image",
29
+ title="CryptoGAN",
30
+ description="These CryptoPunks do not exist. gGnerate your own CryptoPunks").launch()