Update app.py

app.py

@@ -238,6 +238,14 @@ def predict_wsi(image):
 def segment_image(image):
     img = image
     img = np.asarray(img)
+
+    # pad if necessary
+    h, w = img.shape[:2]
+    pad_h = max(0, min_side - h)
+    pad_w = max(0, min_side - w)
+    if(pad_h > 0 or pad_w > 0):
+        img = np.pad(img, ((0, pad_h), (0, pad_w), (0, 0)), mode='constant', constant_values=0)
+    
     if (np.max(img) > 100):
         img = img / 255.0
     transform = T.Compose([T.ToTensor()])