Update app.py

app.py

@@ -25,6 +25,7 @@ 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("After Resize:", image.shape)
@@ -56,9 +57,9 @@ def greet(image):
         print(image.min(), image.max())
         image = image/255.0
         image = image.unsqueeze(0)
+        image = torch.permute(image, [0,3,1,2])
         update_bn(image)
         print(image.shape)
-        image = torch.permute(image, [0,3,1,2])
         out = model(image)
     # model.train()
     return "Hello world!"