Update app.py

app.py

@@ -74,7 +74,7 @@ def show_mask_on_image(raw_image, mask, return_image=False):
         fig = plt.gcf()
         fig.canvas.draw()
         #Convert plot to image
-        img = np.frombuffer(fig.canvas.tostring_argb(), dtype=np.uint8)
+        img = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
         img = img.reshape(fig.canvas.get_width_height()[::-1]+(3,))
         img = Image.fromarray(img)
         plt.close(fig)