leavoigt commited on
Commit
8ca2c43
1 Parent(s): abe2730

Update utils/target_classifier.py

Browse files
Files changed (1) hide show
  1. utils/target_classifier.py +2 -3
utils/target_classifier.py CHANGED
@@ -22,9 +22,8 @@ def get_target_labels(preds):
22
  """
23
 
24
  # Get label names
25
- preds_list = preds.tolist()
26
- st.write('preds_list')
27
- st.write(preds_list)
28
  predictions_names=[]
29
 
30
  # loop through each prediction
 
22
  """
23
 
24
  # Get label names
25
+ st.write("preds - utils")
26
+ st.write(preds[:10])
 
27
  predictions_names=[]
28
 
29
  # loop through each prediction