Update app.py

app.py

@@ -10,11 +10,11 @@ import random
 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):
+  while(train_labels[rn] != digit):
     rn = int(random.random() * 60000)  
-  digit = train_images[rn]
+  digit_img = train_images[rn]
   fig = plt.figure()
-  plt.imshow(digit, cmap=plt.cm.binary)
+  plt.imshow(digit_img, cmap=plt.cm.binary)
   out_txt = "train_images[%d] = %d" % (rn, train_labels[rn])
   return fig, out_txt
 
@@ -25,7 +25,7 @@ iface = gr.Interface(
     #gr.inputs.Number()
     gr.inputs.Slider(0, 9)
   ],
-  outputs=[gr.outputs.Image(type="plot"), 'text'],
+  outputs=['image', 'text'],
   title='MNIST Digit Sampler',
   description='Pick a random digit from the MNIST dataset'
 )