alok94 commited on
Commit
e862f05
·
1 Parent(s): 7a496a2

fixed when sketch is empty and tried to identify

Browse files
Files changed (1) hide show
  1. app.py +18 -1
app.py CHANGED
@@ -92,7 +92,24 @@ def predict_digit(img):
92
  )
93
 
94
  else:
95
- return " "
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
96
 
97
 
98
  # iface=gr.Interface(prdict_digit, inputs='sketchpad', outputs=['label', gr.Slider(0,100, label='Probably 0'), gr.Slider(0,100, label='Probably 1')] ).launch()
 
92
  )
93
 
94
  else:
95
+ simple_empty = pd.DataFrame(
96
+ {
97
+ "a": ["0", "1", "2", "3", "4", "5", "6", "7", "8","9"],
98
+ "b": [0,0,0,0,0,0,0,0,0,0], #[28, 55, 43, 91, 81, 53, 19, 87, 52,80],
99
+ }
100
+ )
101
+
102
+ return " ", gr.BarPlot.update(
103
+ simple_empty,
104
+ x="a",
105
+ y="b",
106
+ x_title="Digits",
107
+ y_title="Identification Probabilities",
108
+ title="Identification Probability",
109
+ tooltip=["a", "b"],
110
+ vertical=False,
111
+ y_lim=[0, 100],
112
+ )
113
 
114
 
115
  # iface=gr.Interface(prdict_digit, inputs='sketchpad', outputs=['label', gr.Slider(0,100, label='Probably 0'), gr.Slider(0,100, label='Probably 1')] ).launch()