Spaces:

joshnguyen
/

mformer

Runtime error

joshnguyen commited on Sep 15, 2023

Commit

ee4b3bb

1 Parent(s): 1d1d7a2

Update label

Files changed (1) hide show

custom_label.py CHANGED Viewed

@@ -119,7 +119,7 @@ class CustomLabel(Changeable, Selectable, IOComponent, JSONSerializable):
             if self.num_top_classes is not None:
                 sorted_pred = sorted_pred[: self.num_top_classes]
             return {
-                "label": None,
                 "confidences": [
                     {"label": pred[0], "confidence": pred[1]} for pred in sorted_pred
                 ],

             if self.num_top_classes is not None:
                 sorted_pred = sorted_pred[: self.num_top_classes]
             return {
+                "label": "",
                 "confidences": [
                     {"label": pred[0], "confidence": pred[1]} for pred in sorted_pred
                 ],