update

app.py

@@ -16,11 +16,9 @@ net_c.load_state_dict(torch.load('./classifier_32.pt')['icnn'])
 net_i.eval()
 net_c.eval()
 def align(x1):
-    x2 = x1
-    h, w = x1.height, x1.width
+    h, w = x1.shape[:2]
     h, w = h // 32 * 32, w // 32 * 32
-    x1 = x1.resize((w, h))
-    x2 = x2.resize((w, h))
+    x1 = np.resize(x1, (h, w, x1.shape[2]))
     return x1
 def predict(img):
     with torch.no_grad():