Update appStore/vulnerability_analysis.py

appStore/vulnerability_analysis.py

@@ -8,7 +8,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
 import streamlit as st
-from utils.indicator_classifier import load_indicatorClassifier, indicator_classification
+from utils.vulnerability_classifier import load_vulnerabilityClassifier, vulnerability_classification
 import logging
 logger = logging.getLogger(__name__)
 from utils.config import get_classifier_params
@@ -59,7 +59,7 @@ def app():
                    
             if 'key1' in st.session_state:
                 df = st.session_state.key1
-                classifier = load_indicatorClassifier(classifier_name=params['model_name'])
+                classifier = load_vulnerabilityClassifier(classifier_name=params['model_name'])
                 st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
 
                 if sum(df['Target Label'] == 'TARGET') > 100:
@@ -67,7 +67,7 @@ def app():
                 else:
                     warning_msg = ""
                     
-                df = indicator_classification(haystack_doc=df,
+                df = vulnerability_classification(haystack_doc=df,
                                             threshold= params['threshold'])
 
                 st.session_state.key1 = df