Update app.py

app.py

@@ -14,7 +14,7 @@ def get_digit(digit_choice):
   digit = train_images[rn]
   fig = plt.figure()
   plt.imshow(digit)
-  out_txt = "digit index: %d\ndigit label: %d" % (digit_choice, rn)
+  out_txt = "digit: %d\ndigit label: %d" % (digit_choice, test_labels[rn])
   return fig, out_txt
 
 iface = gr.Interface(
@@ -22,8 +22,8 @@ iface = gr.Interface(
   #inputs='image',
   #inputs=[gr.inputs.Image(label="Input Image", source="webcam")],
   inputs = [ 
-    gr.inputs.Dropdown([0, 1, 2, 3])
-    #gr.inputs.Number()
+    #gr.inputs.Dropdown([0, 1, 2, 3])
+    gr.inputs.Number()
     #'text'
   ],
   #outputs='image',