update stuff

app.py

@@ -31,11 +31,17 @@ state_dict = torch.load('model.pth', map_location='cpu')
 model.load_state_dict(state_dict, strict=False)
 model.eval()
 
+transform = transforms.Compose([
+    transforms.Resize((28, 28)),
+    transforms.ToTensor(),
+    transforms.Normalize((0.5,), (0.5,))
+])
+
 def predict(image):
     image = image['composite']
-    image = transforms.Resize((28, 28))(image)
+    image = transform(image)
     print(image)
-    
+
     x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.
 
     with torch.no_grad():