Update app.py

app.py

@@ -89,8 +89,10 @@ def plot_outlines(img, masks):
     buf = io.BytesIO()
     fig.savefig(buf, bbox_inches='tight')
     buf.seek(0)
-    output_pil_img = Image.open(buf)
-    return output_pil_img
+    pil_img = Image.open(buf)
+
+    pil_img = pil_img.resize((img.shape[-1], img.shape[-2]), filter = Image.BICUBIC)
+    return pil_img
 
 def plot_overlay(img, masks):
     img = normalize99(img.astype(np.float32).mean(axis=-1))