Update app.py
Browse files
app.py
CHANGED
@@ -106,7 +106,8 @@ def predict_image(img):
|
|
106 |
|
107 |
return pred.item()
|
108 |
|
|
|
109 |
|
110 |
-
iface = gr.Interface(predict_image, inputs=
|
111 |
|
112 |
iface.launch()
|
|
|
106 |
|
107 |
return pred.item()
|
108 |
|
109 |
+
sp = gr.Sketchpad(brush_radius=1.0, shape=(28,28), )
|
110 |
|
111 |
+
iface = gr.Interface(predict_image, inputs=sp, outputs='label', title='Arabic Numbers Recognition', description='Draw a number')
|
112 |
|
113 |
iface.launch()
|