Spaces:
Runtime error
Runtime error
Update app.py
Browse filesCryptoGAN first iteration
app.py
CHANGED
@@ -1,7 +1,30 @@
|
|
1 |
import gradio as gr
|
|
|
|
|
|
|
2 |
|
3 |
-
def greet(name):
|
4 |
-
return "Hello " + name + "!!"
|
5 |
|
6 |
-
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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()
|