Update utils/target_classifier.py

utils/target_classifier.py

@@ -23,7 +23,8 @@ def get_target_labels(preds):
     """
 
     # Turn into list 
-    preds_list = preds.to_list()
+    preds_list = preds.numpy().tolist()
+
     
     # Get label names
     predictions_names=[]