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 |
|