Spaces:

poltextlab
/

babel_machine

Running

kovacsvi commited on 14 days ago

Commit

71aee23

1 Parent(s): 2be2888

oops

Files changed (1) hide show

interfaces/cap_minor.py CHANGED Viewed

@@ -210,7 +210,7 @@ def predict_flat(text, model_id, tokenizer_id, HF_TOKEN=None):
         code = CAP_MIN_NUM_DICT[i]
         prob = probs[i]
-         if code in CAP_LABEL_NAMES:
             # Media (major) topic
             label = CAP_LABEL_NAMES[code]
             display = f"[{code}] {label}"

         code = CAP_MIN_NUM_DICT[i]
         prob = probs[i]
+        if code in CAP_LABEL_NAMES:
             # Media (major) topic
             label = CAP_LABEL_NAMES[code]
             display = f"[{code}] {label}"