Update app.py

app.py

@@ -117,8 +117,8 @@ if 'key0' in st.session_state:
 
         # Text 
         num_paragraphs = len(df_vul['Vulnerability Label'])
-        num_references = len(df_vul[df_vul['Vulnerability Label'] != 'Other'])
-        
+        num_references = df_vul['Vulnerability Label'].apply(lambda x: 'Other' not in x).sum()
+       
         st.markdown(f"""<div style="text-align: justify;"> The document contains a
                 total of <span style="color: red;">{num_paragraphs}</span> paragraphs.
                 We identified <span style="color: red;">{num_references}</span>
@@ -151,7 +151,7 @@ if 'key0' in st.session_state:
         df_label_count = pd.DataFrame(list(group_counts.items()), columns=['Label', 'Count'])
 
         # Merge the label counts with the df_label DataFrame
-        df_label_count = df_labels.merge(df_label_counts, on='Label', how='left')
+        df_label_count = df_labels.merge(df_label_count, on='Label', how='left')
         st.write("df_label_count")
         
     #     # Configure graph