Update utils/target_classifier.py

utils/target_classifier.py

@@ -112,9 +112,12 @@ def target_classification(haystack_doc:pd.DataFrame,
 
         # Get predictions
         predictions = classifier_model(list(haystack_doc.text))
+        st.write("predictions")
+        st.write(predictions)
 
         # Get labels for predictions
-        pred_labels = gettarget_labels(predictions)
+        pred_labels = get_target_labels(predictions)
+        st.write(pred_labels)
 
         # Save labels
         haystack_doc['Target Label'] = pred_labels