Update app.py

app.py

@@ -9,6 +9,7 @@ import numpy as np
 import shutil
 import os
 import openslide
+from project_utils.preprocessing import expand2square
 
 model = UNet(
     spatial_dims=2,
@@ -79,10 +80,10 @@ def process_slide(slide_path):
         slide = openslide.OpenSlide(os.path.join("cache_mrxs", slide_name))
 
     thumbnail = slide.get_thumbnail((512, 512))
-    
-    print(thumbnail.size)
 
-    return thumbnail
+    image = expand2square(thumbnail)
+
+    return process_image(image)
 
 
 interface_slide = gr.Interface(