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 |
],
|