chg: Use output labels to detect output size

This is a more reliable indicator that prediction size

Author: ivyleavedtoadflax

deep_reference_parser/deep_reference_parser.py

@@ -1038,7 +1038,7 @@ def predict(self, X, load_weights=False):
         # If running a single task model, wrap pred_index in a list so that it
         # can use the same logic as multitask models.
 
-        if len(pred_index) == 1 :
+        if len(ind2labelNew) == 1:
             pred_index = [pred_index]
 
         pred_label = []