IMvision12 commited on
Commit
4121dc0
·
1 Parent(s): 2826962

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import numpy as np
7
 
8
  model = from_pretrained_keras("IMvision12/WGAN-GP")
9
 
10
- def Predict(num_images):
11
  random_latent_vectors = tf.random.normal(shape=(int(num_images), 128))
12
  predictions = model.predict(random_latent_vectors)
13
  num = ceil(sqrt(num_images))
@@ -42,4 +42,4 @@ examples = [
42
  ]
43
 
44
 
45
- gr.Interface(Predict, inputs, outputs, title=title, examples=examples, description=description, article=article, analytics_enabled=False).launch()
 
7
 
8
  model = from_pretrained_keras("IMvision12/WGAN-GP")
9
 
10
+ def create_digit_samples(num_images):
11
  random_latent_vectors = tf.random.normal(shape=(int(num_images), 128))
12
  predictions = model.predict(random_latent_vectors)
13
  num = ceil(sqrt(num_images))
 
42
  ]
43
 
44
 
45
+ gr.Interface(create_digit_samples, inputs, outputs, title=title, examples=examples, description=description, article=article, analytics_enabled=False).launch(enable_queue=True)