Update app.py

app.py

@@ -91,7 +91,6 @@ def plot_outlines(img, masks):
     buf.seek(0)
     pil_img = Image.open(buf)
 
-    pil_img = pil_img.resize((img.shape[1], img.shape[0]), resample  = Image.BICUBIC)
     return pil_img
 
 def plot_overlay(img, masks):
@@ -156,6 +155,13 @@ def cellpose_segment(img_input):
 
     overlay = Image.fromarray(overlay)
     flows = Image.fromarray(flows)
+
+    Ly, Lx = img.shape[:2]
+    c = Lx
+    outpix = outpix.resize((Lx, Ly), resample  = Image.BICUBIC)
+    overlay = overlay.resize((Lx, Ly), resample  = Image.BICUBIC)
+    flows = flows.resize((Lx, Ly), resample  = Image.BICUBIC)
+
     #masks = Image.fromarray(255. * crand[masks])
 
     pil_masks = Image.fromarray(masks.astype('int32'))
@@ -202,9 +208,9 @@ with gr.Blocks(title = "Hello",
 
 
         with gr.Column(scale=2):            
-            img_outlines = gr.Image(label = "Outlines", type = "pil", format = 'png', width = "50vw", height = "20vw")
-            img_overlay = gr.Image(label = "Overlay", type = "pil", format = 'png', width = "50vw", height = "20vw")
-            flows = gr.Image(label = "Cellpose flows", type = "pil", format = 'png', width = "50vw", height = "20vw")
+            img_outlines = gr.Image(label = "Outlines", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
+            img_overlay = gr.Image(label = "Overlay", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
+            flows = gr.Image(label = "Cellpose flows", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
             #masks = gr.Image(label = "Output image", type = "numpy")