Update app.py

app.py

@@ -7,7 +7,7 @@ import numpy as np
 
 model = from_pretrained_keras("IMvision12/WGAN-GP")
 
-def create_digit_samples(num_images):
+def Predict(num_images):
       random_latent_vectors = tf.random.normal(shape=(int(num_images), 128))
       predictions = model.predict(random_latent_vectors)
       num = ceil(sqrt(num_images))
@@ -42,4 +42,4 @@ examples = [
 ]
 
 
-gr.Interface(create_digit_samples, inputs, outputs, title=title, examples=examples, description=description, article=article, analytics_enabled=False).launch(debug=True)
+gr.Interface(Predict, inputs, outputs, title=title, examples=examples, description=description, article=article, analytics_enabled=False).launch(debug=True)