Spaces:
Sleeping
Sleeping
Update appStore/target.py
Browse files- 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[
|
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'])
|