Update utils/target_classifier.py

utils/target_classifier.py

@@ -103,17 +103,24 @@ def target_classification(haystack_doc:pd.DataFrame,
     logging.info("Working on target/action identification")
                             
     haystack_doc['Target Label'] = 'NA'
-                           
+    st.write("haystack_doc")
+    st.write(haystack_doc)
+                            
     if not classifier_model:
 
-        classifier_model = st.session_state['target_classifier']
+        st.write("No classifier_model")
 
+        classifier_model = st.session_state['target_classifier']
+        st.write("classifier model defined")
+        
         # Get predictions
         predictions = classifier_model(list(haystack_doc.text))
-
+        st.write("predictions made")
+        st.predictions.head(20)
         # Get labels for predictions
         pred_labels = get_target_labels(predictions)
-
+        st.write("pred_labels")
+        st.write(pred_labels)
         # Save labels
         haystack_doc['Target Label'] = pred_labels