Update app.py

app.py

@@ -211,8 +211,8 @@ def predict_wsi(image):
     num_classes=5
     pred_labels = torch.zeros(1, num_classes+1, image.shape[2], image.shape[3]).cuda()
     counter_tensor = torch.zeros(1, 1, image.shape[2], image.shape[3]).cuda()
-    for i in range(0, image.shape[2] - patch_size, stride):
-        for j in range(0, image.shape[3] - patch_size, stride):
+    for i in range(0, image.shape[2] - patch_size + stride, stride):
+        for j in range(0, image.shape[3] - patch_size + stride, stride):
             i_lowered = min(i, image.shape[2] - patch_size)
             j_lowered = min(j, image.shape[3] - patch_size)
             patch = image[:, :, i_lowered:i_lowered + patch_size, j_lowered:j_lowered + patch_size]