Update app.py

app.py

@@ -12,10 +12,16 @@ import re
 import matplotlib.pyplot as plt
 from io import BytesIO
 
+# 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]
+# )
+
 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]
+    mean=[0.49139968, 0.48215827 ,0.44653124],
+    std=[0.24703233, 0.24348505, 0.26158768]
 )
+
 classes = ('plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck')
 
 model = LITResNet(classes)