Update utils/target_classifier.py

utils/target_classifier.py

@@ -23,7 +23,8 @@ def get_target_labels(preds):
     
     # Get label names
     preds_list = preds.tolist()
-    
+    st.write('preds_list')
+    st.write(preds_list)
     predictions_names=[]
     
     # loop through each prediction
@@ -112,13 +113,13 @@ def target_classification(haystack_doc:pd.DataFrame,
 
         # Get predictions
         predictions = classifier_model(list(haystack_doc.text))
-        st.write("predictions")
-        st.write(predictions)
+        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')
+        st.write(pred_labels[:10])
         # Save labels
         haystack_doc['Target Label'] = pred_labels