Spaces:

chrisjay
/

simple-mnist-classification

Runtime error

chrisjay commited on Jul 4, 2022

Commit

2b63af2

1 Parent(s): ba1e284

initial commit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -112,7 +112,7 @@ def main():
             image_input =gr.inputs.Image(source="canvas",shape=(28,28),invert_colors=True,image_mode="L",type="pil")
-            label_output = gr.outputs.Label(num_top_classes=2)
         image_input.change(image_classifier,inputs = [image_input],outputs=[label_output])

             image_input =gr.inputs.Image(source="canvas",shape=(28,28),invert_colors=True,image_mode="L",type="pil")
+            label_output = gr.outputs.Label(num_top_classes=10)
         image_input.change(image_classifier,inputs = [image_input],outputs=[label_output])