add legend

app.py

@@ -40,8 +40,8 @@ state = resume(
 daily_model = daily_model.eval()
 annual_model = annual_model.eval()
 
-st.title('Sentinel 2 Crop Mapping')
-st.markdown('Using a <model> and giving a zip that contains 32 tiff named correctly you can reach prediction of crop mapping og the area.')
+st.title('Lombardia Sentinel 2 Crop Mapping')
+st.markdown('Using a daily FPN and giving a zip that contains 30 tiff named correctly you can reach prediction of crop mapping og the area.')
 
 file_uploaded = st.file_uploader(
     "Upload",
@@ -164,4 +164,14 @@ if st.paths is not None:
 
     fig, ax = plt.subplots()
     ax.imshow(target)
-    st.pyplot(fig)
+    
+
+    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>'
+
+    col1, col2 = st.columns(2)
+    with col1:
+        st.pyplot(fig)
+    with col2:
+        st.markdown(markdown_legend, unsafe_allow_html=True)

utils.py

@@ -760,3 +760,4 @@ classes_color_map = np.array([[0., 0., 0.],                          # Black
                               [0.46666667, 0.04313725, 0.1254902],
                               [0., 0., 0.55686275]])
 colors = classes_color_map[:8] # num classes
+labels_map = ['Unknown', 'Other cereals', 'Woods and other tree crops', 'Forage', 'Corn', 'Rice', 'Unknow cropland', 'No arable land']