Show data

app.py

@@ -47,6 +47,9 @@ def download_comparison_images():
 
 download_comparison_images()
 
+# initialize prediction visual data
+#coco_df = 
+
 if "output_1" not in st.session_state:
     img_1 = Image.open('plant_pid.png')
     st.session_state["output_1"] = img_1.resize((4960,3508))
@@ -55,6 +58,9 @@ if "output_2" not in st.session_state:
     img_2 = Image.open('prediction_visual.png')
     st.session_state["output_2"] = img_2.resize((4960,3508))
 
+#if "output_3" not in st.session_state:
+    
+
 col1, col2, col3 = st.columns(3, gap='medium')
 with col1:
     with st.expander('How to use it'):
@@ -167,6 +173,22 @@ with col2:
         with tab2:
             st.image(st.session_state["output_2"])
         with tab3:
-            st.dataframe(coco_df)
+            st.dataframe(
+                coco_df,
+                column_config = {
+                    'category' : 'Category',
+                    'score' : 'Confidence',
+                },
+                hide_index = True,
+            )
+            st.dataframe(
+                output_df,
+                column_config = {
+                    'category' : 'Category',
+                    'count' : 'Number of Elements',
+                    'percentage' : '%',
+                },
+                hide_index = True,
+            )