Spaces:

crobbi
/

MNIST-Digit-Classifier

Runtime error

crobbi commited on Sep 3, 2023

Commit

edb390c

1 Parent(s): f02d2b7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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