Spaces:
Sleeping
Sleeping
Update appStore/target.py
Browse files- appStore/target.py +3 -0
appStore/target.py
CHANGED
@@ -62,6 +62,9 @@ def app():
|
|
62 |
# Load the existing dataset
|
63 |
df = st.session_state.key1
|
64 |
|
|
|
|
|
|
|
65 |
# Load the classifier model
|
66 |
classifier = load_targetClassifier(classifier_name=params['model_name'])
|
67 |
|
|
|
62 |
# Load the existing dataset
|
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'])
|
70 |
|