Update app.py

app.py

@@ -17,26 +17,26 @@ article = """
         </p>
         """
 
-def create_digit_samples(n_samples):
-        random_latent_vectors = tf.random.normal(shape=(int(n_samples), 128))
-        examples = model(random_latent_vectors) 
-        size = ceil(sqrt(n_samples))
-        digit_images = np.zeros((28*size, 28*size), dtype=float)
+def Predict(num_images):
+        random_latent_vectors = tf.random.normal(shape=(int(num_images), 128))
+        preds = model(random_latent_vectors) 
+        num = ceil(sqrt(num_images))
+        images = np.zeros((28*num, 28*num), dtype=float)
         n = 0
-        for i in range(size):
-            for j in range(size):
-                if n == n_samples:
+        for i in range(num):
+            for j in range(num):
+                if n == num_images:
                     break
-                digit_images[i* 28 : (i+1)*28, j*28 : (j+1)*28] = examples[n, :, :, 0]
+                images[i* 28 : (i+1)*28, j*28 : (j+1)*28] = preds[n, :, :, 0]
                 n += 1
-        return digit_images
+        return images
 
 
 examples = [[5],[8],[2],[3]]
 
 
 iface = gr.Interface(
-    fn = create_digit_samples,
+    fn = Predict,
     inputs = ["number"],
     outputs = ["image"],
     examples = examples,