Update appStore/vulnerability_analysis.py

appStore/vulnerability_analysis.py

@@ -72,6 +72,10 @@ def app():
                 df = vulnerability_classification(haystack_doc=df,
                                             threshold= params['threshold'])
 
+                # Filter the dataframe to only show the paragraphs with references
+                df_filtered = df[df['Vulnerability Label'].apply(lambda x: 'Other' not in x)]
+
+
                 # Store df in session state with key1
                 st.session_state.key1 = df