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'])
|