update app.py

app.py

@@ -43,6 +43,12 @@ model = LitResnet(lr=0.05).load_from_checkpoint("weights_92.ckpt")
 
 device = torch.device("cpu")
 
+# Denormalize the data using test mean and std deviation
+inv_normalize = transforms.Normalize(
+    mean=[-0.50/0.23, -0.50/0.23, -0.50/0.23],
+    std=[1/0.23, 1/0.23, 1/0.23]
+)
+
 # Get the misclassified data from test dataset
 misclassified_data = get_misclassified_data2(model, device, 20)