Spaces:
Runtime error
Runtime error
Commit
·
1d1d7a2
1
Parent(s):
508ada6
Update sorting mechanism
Browse files- custom_label.py +2 -2
custom_label.py
CHANGED
@@ -115,11 +115,11 @@ class CustomLabel(Changeable, Selectable, IOComponent, JSONSerializable):
|
|
115 |
y = y["confidences"]
|
116 |
y = {c["label"]: c["confidence"] for c in y}
|
117 |
sorted_pred = sorted(
|
118 |
-
y.items(), key=operator.itemgetter(0), reverse=
|
119 |
if self.num_top_classes is not None:
|
120 |
sorted_pred = sorted_pred[: self.num_top_classes]
|
121 |
return {
|
122 |
-
"label":
|
123 |
"confidences": [
|
124 |
{"label": pred[0], "confidence": pred[1]} for pred in sorted_pred
|
125 |
],
|
|
|
115 |
y = y["confidences"]
|
116 |
y = {c["label"]: c["confidence"] for c in y}
|
117 |
sorted_pred = sorted(
|
118 |
+
y.items(), key=operator.itemgetter(0), reverse=False)
|
119 |
if self.num_top_classes is not None:
|
120 |
sorted_pred = sorted_pred[: self.num_top_classes]
|
121 |
return {
|
122 |
+
"label": None,
|
123 |
"confidences": [
|
124 |
{"label": pred[0], "confidence": pred[1]} for pred in sorted_pred
|
125 |
],
|