updating overlay

app.py

@@ -49,11 +49,14 @@ def plot_outlines(img, masks):
     bytes_image = io.BytesIO()
     plt.savefig(bytes_image, format='png', facecolor=fig.get_facecolor(), edgecolor='none')
     bytes_image.seek(0)
-    img_html = base64.b64encode(bytes_image.getvalue()).decode()
+    img_arr = np.frombuffer(bytes_image.getvalue(), dtype=np.uint8)
+    bytes_image.close()
+    img = cv2.imdecode(img_arr, 1)
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     del bytes_image
     fig.clf()
     plt.close(fig)
-    return img_html
+    return img
 
 def plot_overlay(img, masks):
     img = normalize99(img.astype(np.float32).mean(axis=-1))