Spaces:
Running
Running
fix dict
Browse files- 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 |
-
|
|
|
|
|
|
|
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 |
|