Update app.py

app.py

@@ -11,7 +11,7 @@ def test(k):
         print(k.shape)
         k = k / 255.
         model = create_model()
-        return np.argmax(model.predict(k))
+        return np.argmax(model.predict(k.reshape(-1,28,28)))
 gr.Interface(inputs=gr.Sketchpad(),
             outputs="number",
             fn=test,