Spaces:
Running
Running
handle 999 case
Browse files- interfaces/cap_minor.py +4 -0
interfaces/cap_minor.py
CHANGED
@@ -37,6 +37,10 @@ def convert_minor_to_major(results, probs):
|
|
37 |
for i in results:
|
38 |
prob = probs[i]
|
39 |
major_code = str(CAP_MIN_NUM_DICT[i])[:-2]
|
|
|
|
|
|
|
|
|
40 |
label = CAP_LABEL_NAMES[int(major_code)]
|
41 |
|
42 |
key = f"[{major_code}] {label}"
|
|
|
37 |
for i in results:
|
38 |
prob = probs[i]
|
39 |
major_code = str(CAP_MIN_NUM_DICT[i])[:-2]
|
40 |
+
|
41 |
+
if major_code == "99":
|
42 |
+
major_code = "999"
|
43 |
+
|
44 |
label = CAP_LABEL_NAMES[int(major_code)]
|
45 |
|
46 |
key = f"[{major_code}] {label}"
|