Update app.py

app.py

@@ -24,9 +24,10 @@ def print_bn():
 
 def update_bn(image):
     cursor_im = 0
+    print("Before Resize:", image.shape)
     image = T.Resize((40,40))(image)
     image = image.reshape(-1)
-    print(image.shape)
+    print("After Resize:", image.shape)
     for m in model.modules():
         if(type(m) is nn.BatchNorm2d):
             if(cursor_im < image.shape[0]):