change plotting

app.py

@@ -49,9 +49,11 @@ def FRFsegment(input_img):
     pred = resize(mask, (w, h), preserve_range=True, clip=True)
     
     #overlay plot
-    p = plt.imshow(input_img,cmap='gray')
-    p = plt.imshow(pred, alpha=0.4)
-    p = plt.axis("off")
+    plt.clf()
+    plt.imshow(input_img,cmap='gray')
+    plt.imshow(pred, alpha=0.4)
+    plt.axis("off")
+    plt.margins(x=0, y=0)
     return plt