colors hex

app.py

@@ -4,6 +4,7 @@ import zipfile
 import torch
 from utils import *
 import matplotlib.pyplot as plt
+from matplotlib import colors
 
 if not hasattr(st, 'paths'):
     st.paths = None
@@ -168,7 +169,8 @@ if st.paths is not None:
 
     markdown_legend = ''
     for c, l in zip(classes_color_map, labels_map):
-        markdown_legend += f'<div style="color:gray;background-color: rgb({c[0]*255}, {c[1]*255}, {c[2]*255});">{l}</div><br>'
+        print(colors.to_hex(c))
+        markdown_legend += f'<div style="color:gray;background-color: {colors.to_hex(c)};">{l}</div><br>'
 
     col1, col2 = st.columns(2)
     with col1: