Spaces:

poltextlab
/

babel_machine

Running

poltextlab commited on May 5

Commit

b6f6b98

verified ·

1 Parent(s): a115642

fix

Files changed (1) hide show

interfaces/cap_minor.py CHANGED Viewed

@@ -36,7 +36,7 @@ def convert_minor_to_major(results, probs):
     results_as_text = dict()
     for i in results:
         prob = probs[i]
-        major_code = CAP_MIN_NUM_DICT[i][:-2]
         label = CAP_LABEL_NAMES[major_code]
         key = f"[{major_code}] {label}"

     results_as_text = dict()
     for i in results:
         prob = probs[i]
+        major_code = str(CAP_MIN_NUM_DICT[i])[:-2]
         label = CAP_LABEL_NAMES[major_code]
         key = f"[{major_code}] {label}"