Spaces:
Running
Running
fix types
Browse files- interfaces/cap_minor.py +1 -1
interfaces/cap_minor.py
CHANGED
@@ -37,7 +37,7 @@ 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[major_code]
|
41 |
|
42 |
key = f"[{major_code}] {label}"
|
43 |
results_as_text[key] = probs[i]
|
|
|
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}"
|
43 |
results_as_text[key] = probs[i]
|