Update app.py

app.py

@@ -22,7 +22,7 @@ model.load_state_dict(torch.load("./model_best.pt?dl=0", map_location=device))
 # from torchvision import transforms
 
 transforms2 =  transforms.Compose([
-        transforms.Resize(256),
+        transforms.Resize(224),
         transforms.ToTensor(),
         transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
     ])
@@ -59,7 +59,7 @@ def recognize_digit(image):
 
 
 im = gradio.inputs.Image(
-    shape=(256, 256), image_mode="RGB", type="pil")
+    shape=(224, 224), image_mode="RGB", type="pil")
 
 iface = gr.Interface(
     recognize_digit,