Update app.py

app.py

@@ -237,18 +237,19 @@ def predict_wsi(image):
 @spaces.GPU(duration=120)
 def segment_image(image):
     img = image
+    img = np.asarray(img)
 
     # resize if necessary
-    # min_side=768
-    # print(img.size)
-    # w, h = img.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)
+    h, w = img.shape[:2]
+    if min(h, w) < min_side:
+        scale = min_side / min(h, w)
+        new_w, new_h = int(w * scale), int(h * scale)
+        # Convert NumPy array to PIL Image
+        image = Image.fromarray(img)
+        # Resize the image using PIL
+        resized_image = image.resize((new_w, new_h), Image.ANTIALIAS)
+        img = np.array(resized_image)
     
-    img = np.asarray(img)
-    print(img.shape)
     if (np.max(img) > 100):
         img = img / 255.0
     transform = T.Compose([T.ToTensor()])