Update app.py

app.py

@@ -18,7 +18,7 @@ article = """
         </p>
         """
 
-def create_digit_samples(model, num_images):
+def Predict(model, num_images):
       random_latent_vectors = tf.random.normal(shape=(int(num_images), 128))
       predictions = model(random_latent_vectors)
       num = ceil(sqrt(num_images))
@@ -33,14 +33,14 @@ def create_digit_samples(model, num_images):
       return images
 
 def inference(num_images, select: str):
-    if select == 'fmnist':
-        result = create_digit_samples(model2, num_images)
+    if select == 'Fashion_mnist':
+        result = Predict(model2, num_images)
     else:
-        result = create_digit_samples(model1, num_images)
+        result = Predict(model1, num_images)
     return result
 
-inputs = [gr.inputs.Number(label="number of images"), gr.inputs.Radio(['fmnist', 'mnist'])]
+inputs = [gr.inputs.Number(label="number of images"), gr.inputs.Radio(['Fashion_mnist', 'Handwritten_digits_mnist'])]
 outputs = gr.outputs.Image(label="Output Image")
     
 
-gr.Interface(inference, inputs, outputs, analytics_enabled=False).launch(share=True)
+gr.Interface(inference, inputs, outputs, title=title, description=description, article=article, analytics_enabled=False).launch(share=True)