Update appStore/target.py

appStore/target.py

@@ -24,8 +24,10 @@ from pandas.api.types import (
     is_list_like)
 
 # Declare all the necessary variables
-classifier_identifier = 'tapp'
-params  = get_classifier_params(classifier_identifier)
+tapp_classifier_identifier = 'tapp'
+target_setfit = 'target'
+param1  = get_classifier_params(tapp_classifier_identifier)
+param2  = get_classifier_params(target_setfit)
 
 def app():
     ### Main app code ###
@@ -34,8 +36,10 @@ def app():
             df = st.session_state.key0
 
             #load Classifiers
-            classifier = load_tappClassifier(classifier_name=params['model_name'])
-            st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
+            classifier = load_tappClassifier(classifier_name=param1['model_name'])
+            st.session_state['{}_classifier'.format(tapp_classifier_identifier)] = classifier
+            classifier = load_targetClassifier(classifier_name=param2['model_name'])
+            st.session_state['{}_classifier'.format(target_setfit)] = classifier
             
             if len(df) > 100:
                 warning_msg = ": This might take sometime, please sit back and relax."