Dimitre commited on
Commit
047136e
·
1 Parent(s): e29d124

Update app.py

Browse files

Adding seed input

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,8 +12,8 @@ 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))
@@ -24,7 +24,7 @@ def generate():
24
  return fig
25
 
26
  gr.Interface(fn=generate,
27
- inputs="text",
28
  outputs="image",
29
  title="CryptoGAN",
30
  description="These CryptoPunks do not exist. gGnerate your own CryptoPunks").launch()
 
12
  image = ((image + 1) / 2) # scale back
13
  return image
14
 
15
+ def generate(seed):
16
+ noise = tf.random.normal(shape=[n_images, codings_size], seed=seed)
17
  generated_images = hub_generator(noise, training=False)
18
 
19
  fig = plt.figure(figsize=(10, 10))
 
24
  return fig
25
 
26
  gr.Interface(fn=generate,
27
+ inputs=[gr.inputs.Slider(label='Seed', minimum=0, maximum=1000, default=42)],
28
  outputs="image",
29
  title="CryptoGAN",
30
  description="These CryptoPunks do not exist. gGnerate your own CryptoPunks").launch()