Update app.py

app.py

@@ -53,10 +53,10 @@ def greet(image):
         print(image.min(), image.max())
         image = image/255.0
         image = image.unsqueeze(0)
+        update_bn(image)
         print(image.shape)
         image = torch.permute(image, [0,3,1,2])
         out = model(image)
-        update_bn(image)
     # model.train()
     return "Hello world!"