leavoigt commited on
Commit
2f12850
1 Parent(s): 329d6cf

Update appStore/target.py

Browse files
Files changed (1) hide show
  1. appStore/target.py +1 -1
appStore/target.py CHANGED
@@ -63,7 +63,7 @@ def app():
63
  df = st.session_state.key1
64
 
65
  # Filter out all paragraphs that do not have a reference to groups
66
- df = df[~df['Vulnerability Label'].apply(lambda labels: 'Other' in labels)]
67
 
68
  # Load the classifier model
69
  classifier = load_targetClassifier(classifier_name=params['model_name'])
 
63
  df = st.session_state.key1
64
 
65
  # Filter out all paragraphs that do not have a reference to groups
66
+ df = df[df['Vulnerability Label'].apply(lambda x: len(x) > 0 and 'Other' not in x)]
67
 
68
  # Load the classifier model
69
  classifier = load_targetClassifier(classifier_name=params['model_name'])