Update app.py

app.py

@@ -7,7 +7,7 @@ import random
 
 (train_images, train_labels), (test_images, test_labels) = mnist.load_data()
 
-def pick_digit(digit_choice):
+def sample_digit(digit):
   rn = 0
   # pick a random digit from 60,000 in the training set until a desired match is found
   while(train_labels[rn] != digit_choice):
@@ -19,14 +19,15 @@ def pick_digit(digit_choice):
   return fig, out_txt
 
 iface = gr.Interface(
-  fn = pick_digit,
-  inputs = [ 
+  fn = sample_digit,
+  inputs = ('digit', [ 
     #gr.inputs.Dropdown([0, 1, 2, 3])
-    gr.inputs.Number()
-  ],
+    #gr.inputs.Number()
+    gr.inputs.Slider(0, 9)
+  ]),
   outputs=[gr.outputs.Image(type="plot"), 'text'],
-  title='page title',
-  description='page description'
+  title='MNIST Digit Sampler',
+  description='Pick a random digit from the MNIST dataset'
 )
   
 iface.launch()