Update app.py

app.py

@@ -237,16 +237,16 @@ def predict_wsi(image):
 @spaces.GPU(duration=120)
 def segment_image(image):
     img = image
-    img = np.asarray(img)
-
-    # pad if necessary
-    min_side = 768
-    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)
+
+    # resize if necessary
+    min_side=768
+    w, h = image.size
+    if min(w, h) < min_side:
+        scale = min_side / min(w, h)
+        new_w, new_h = int(w * scale), int(h * scale)
+        img = img.resize((new_w, new_h), Image.ANTIALIAS)
     
+    img = np.asarray(img)    
     if (np.max(img) > 100):
         img = img / 255.0
     transform = T.Compose([T.ToTensor()])