Update app.py

app.py

@@ -113,13 +113,13 @@ if 'key0' in st.session_state:
             # Create a df that stores how often the labels appear
             df_count = pd.DataFrame(list(label_dict.items()), columns=['Label ID', 'Label'])
 
-            # Count how often each label appears in the "Vulnerability Labels" column
-            label_counts = st.session_state['key0']['Vulnerability Labels'].value_counts().reset_index()
-            label_counts.columns = ['Label', 'Count']
+            # # Count how often each label appears in the "Vulnerability Labels" column
+            # label_counts = st.session_state['key0']['Vulnerability Labels'].value_counts().reset_index()
+            # label_counts.columns = ['Label', 'Count']
 
-            # Merge the label counts with the df_label DataFrame
-            df_label = df_label.merge(label_counts, on='Label', how='left')
-            st.write(df_label)
+            # # Merge the label counts with the df_label DataFrame
+            # df_label = df_label.merge(label_counts, on='Label', how='left')
+            # st.write(df_label)
             #bar_data = st.session_state['key0']