Spaces:

TeresaK
/

cpv_test

Runtime error

leavoigt commited on Jan 5

Commit

69fa6c8

•

1 Parent(s): bf7f98d

Update utils/target_classifier.py

Files changed (1) hide show

utils/target_classifier.py CHANGED Viewed

@@ -116,7 +116,7 @@ def target_classification(haystack_doc:pd.DataFrame,
         # Get predictions
         predictions = classifier_model(list(haystack_doc.text))
         st.write("predictions made")
-        st.write(predictions.head(20))
         # Get labels for predictions
         pred_labels = get_target_labels(predictions)
         st.write("pred_labels")

         # Get predictions
         predictions = classifier_model(list(haystack_doc.text))
         st.write("predictions made")
+        st.write(predictions)
         # Get labels for predictions
         pred_labels = get_target_labels(predictions)
         st.write("pred_labels")