IMvision12 commited on
Commit
d81477d
·
1 Parent(s): f03cc9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,7 +17,7 @@ article = """
17
  </p>
18
  """
19
 
20
- def create_digit_samples(num_images):
21
  random_latent_vectors = tf.random.normal(shape=(int(num_images), 128))
22
  predictions = model.predict(random_latent_vectors)
23
  num = ceil(sqrt(num_images))
@@ -44,4 +44,4 @@ examples = [
44
  ]
45
 
46
 
47
- gr.Interface(create_digit_samples, inputs, outputs, article=article, title=title, description=description, examples=examples, analytics_enabled=False).launch(enable_queue=True)
 
17
  </p>
18
  """
19
 
20
+ def Predict(num_images):
21
  random_latent_vectors = tf.random.normal(shape=(int(num_images), 128))
22
  predictions = model.predict(random_latent_vectors)
23
  num = ceil(sqrt(num_images))
 
44
  ]
45
 
46
 
47
+ gr.Interface(Predict, inputs, outputs, article=article, title=title, description=description, examples=examples, analytics_enabled=False).launch(enable_queue=True)