Spaces:

Gaimundo
/

mnist-cnn-space

Sleeping

Gaimundo commited on Oct 9

Commit

a55ad39

verified ·

1 Parent(s): 8336709

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ def predict(image):
 iface = gr.Interface(
     fn=predict,
-    inputs=gr.Image(type="pil", shape=(28,28), image_mode="L", label="Draw a digit"),
     outputs=gr.Label(num_top_classes=10),
     title="MNIST Digit Classifier (PyTorch)"
 )

 iface = gr.Interface(
     fn=predict,
+    inputs=gr.Image(type="pil", image_mode="L", label="Draw a digit"),
     outputs=gr.Label(num_top_classes=10),
     title="MNIST Digit Classifier (PyTorch)"
 )