c-walls commited on
Commit
d488429
·
1 Parent(s): 02c4f29

fix dict indexing

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ import tensorflow as tf
6
  def mnist_classifier(img):
7
  #img = tf.image.resize(img, [28, 28])
8
  #img = tf.cast(img, tf.float32)
9
- print(img)
10
- return "returned: " + img
11
 
12
  demo = gr.Interface(fn=mnist_classifier, inputs="sketchpad", outputs="text", title="MNIST Checker", description="Draw a number 0-9 to see if the model can classify it.")
13
  demo.launch()
 
6
  def mnist_classifier(img):
7
  #img = tf.image.resize(img, [28, 28])
8
  #img = tf.cast(img, tf.float32)
9
+ print(img[0])
10
+ return "returned: " + img[0]
11
 
12
  demo = gr.Interface(fn=mnist_classifier, inputs="sketchpad", outputs="text", title="MNIST Checker", description="Draw a number 0-9 to see if the model can classify it.")
13
  demo.launch()