Update utils/target_classifier.py

utils/target_classifier.py

@@ -27,7 +27,7 @@ def get_target_labels(preds):
     predictions_names=[]
     
     # loop through each prediction
-    for ele in preds_list:
+    for ele in preds:
     
       # see if there is a value 1 and retrieve index
       try:
@@ -105,18 +105,15 @@ def target_classification(haystack_doc:pd.DataFrame,
     logging.info("Working on target/action identification")
                             
     haystack_doc['Target Label'] = 'NA'
-
-    ## TEST DELETE ###
-    if classifier_model: 
-        st.write("This is the classifier model")
-        st.write(classifier_model)
-                            
+                           
     if not classifier_model:
 
-        target_classifier_model = st.session_state['target_classifier']
+        classifier_model = st.session_state['target_classifier']
+        st.write("utils - classifier model")
+        st.write(classifier_model)
 
         # Get predictions
-        predictions = target_classifier_model(list(haystack_doc.text))
+        predictions = classifier_model(list(haystack_doc.text))
         st.write('predictions')
         st.write(predictions[:10])