add img_to_bytes function

ahmedbesbes · Apr 1, 2021 · 55fd2c1 · 55fd2c1
1 parent 50a5dde
commit 55fd2c1
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/utils/functions.py b/utils/functions.py
@@ -1,3 +1,5 @@
+from pathlib import Path
+import base64
 import time
 import streamlit as st
 import pandas as pd
@@ -236,4 +238,10 @@ def display_kpis(train_metric, test_metric, label):
         margin=dict(l=0, r=0, b=5, t=50, pad=0),
     )
 
-    return fig
+    return fig
+
+
+def img_to_bytes(img_path):
+    img_bytes = Path(img_path).read_bytes()
+    encoded = base64.b64encode(img_bytes).decode()
+    return encoded