Update app.py

app.py

@@ -34,6 +34,7 @@ load_model.eval()
 def recognize_digit(image):
     if image is not None:
         # Preprocess of the image
+        image = Image.fromarray(image).resize((28, 28))
         transform = transforms.Compose([
                         transforms.ToTensor(),
                         transforms.Normalize((0.5,), (0.5,))
@@ -48,6 +49,6 @@ def recognize_digit(image):
     
 
 demo = gr.Interface(fn=recognize_digit, 
-                    inputs=gr.Image(shape=(28,28), image_mode="L", invert_colors=True, source="canvas"), 
+                    inputs=gr.Image(image_mode="L", invert_colors=True, source="canvas"), 
                     outputs=gr.Label(num_top_classes=1))
 demo.launch(True)