Update utils/target_classifier.py

utils/target_classifier.py

@@ -107,18 +107,13 @@ def target_classification(haystack_doc:pd.DataFrame,
     if not classifier_model:
 
         classifier_model = st.session_state['target_classifier']
-        st.write("utils - classifier model")
-        st.write(classifier_model)
 
         # Get predictions
         predictions = classifier_model(list(haystack_doc.text))
-        st.write('predictions')
-        st.write(predictions[:10])
 
         # Get labels for predictions
         pred_labels = get_target_labels(predictions)
-        st.write('pred_labels')
-        st.write(pred_labels[:10])
+
         # Save labels
         haystack_doc['Target Label'] = pred_labels