Update App_main.py

App_main.py

@@ -13,7 +13,7 @@ device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
 
 def construct_sample(img, mean=0.5, std=0.5):
     img = transforms.ToTensor()(img)
-    img = transforms.Resize(48, InterpolationMode.BICUBIC)(img)
+    img = transforms.Resize((48, 48), InterpolationMode.BICUBIC)(img)
     img = transforms.Normalize(mean, std)(img)
     return img