poltextlab commited on
Commit
20681cd
·
verified ·
1 Parent(s): af15b30
Files changed (1) hide show
  1. interfaces/cap_minor.py +4 -1
interfaces/cap_minor.py CHANGED
@@ -44,7 +44,10 @@ def convert_minor_to_major(results, probs):
44
  label = CAP_LABEL_NAMES[int(major_code)]
45
 
46
  key = f"[{major_code}] {label}"
47
- results_as_text[key] += probs[i]
 
 
 
48
 
49
  return results_as_text
50
 
 
44
  label = CAP_LABEL_NAMES[int(major_code)]
45
 
46
  key = f"[{major_code}] {label}"
47
+ if key in results_as_text:
48
+ results_as_text[key] += probs[i]
49
+ else:
50
+ results_as_text[key] = probs[i]
51
 
52
  return results_as_text
53