alexluna4 commited on
Commit
7174bff
·
verified ·
1 Parent(s): 9519c18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -45,9 +45,10 @@ def recognize_digit(image):
45
  return {str(i): float(prediction[0][i]) for i in range(10)}
46
  else:
47
  return ""
48
-
49
 
 
50
  demo = gr.Interface(fn=recognize_digit,
51
- inputs=gr.Image(width=28,height=28,image_mode="L", invert_colors=True, source="canvas"),
52
- outputs=gr.Label(num_top_classes=1))
 
53
  demo.launch(True)
 
45
  return {str(i): float(prediction[0][i]) for i in range(10)}
46
  else:
47
  return ""
 
48
 
49
+
50
  demo = gr.Interface(fn=recognize_digit,
51
+ inputs="sketchpad",
52
+ outputs=gr.Label(num_top_classes=1),
53
+ live=True)
54
  demo.launch(True)