Spaces:

poltextlab
/

babel_machine

Starting

poltextlab commited on 28 days ago

Commit

af15b30

verified ·

1 Parent(s): 5c48f48

cumulate probabilities

Files changed (1) hide show

interfaces/cap_minor.py CHANGED Viewed

@@ -44,7 +44,7 @@ def convert_minor_to_major(results, probs):
         label = CAP_LABEL_NAMES[int(major_code)]
         key = f"[{major_code}] {label}"
-        results_as_text[key] = probs[i]
     return results_as_text

         label = CAP_LABEL_NAMES[int(major_code)]
         key = f"[{major_code}] {label}"
+        results_as_text[key] += probs[i]
     return results_as_text