poltextlab commited on
Commit
a552e92
·
verified ·
1 Parent(s): b6f6b98
Files changed (1) hide show
  1. 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]