kovacsvi commited on
Commit
0f6a4f6
·
1 Parent(s): 882055b
Files changed (1) hide show
  1. interfaces/cap_minor_media.py +1 -1
interfaces/cap_minor_media.py CHANGED
@@ -179,7 +179,7 @@ def predict_flat(text, model_id, tokenizer_id, HF_TOKEN=None):
179
  # Minor topic
180
  major_code = code // 100
181
  major_label = CAP_MEDIA_LABEL_NAMES[major_code]
182
- minor_label = CAP_LABEL_NAMES[code]
183
  display = f"[{major_code}] {major_label} [{code}] {minor_label}"
184
 
185
  output_pred[display] = prob
 
179
  # Minor topic
180
  major_code = code // 100
181
  major_label = CAP_MEDIA_LABEL_NAMES[major_code]
182
+ minor_label = CAP_MIN_LABEL_NAMES[code]
183
  display = f"[{major_code}] {major_label} [{code}] {minor_label}"
184
 
185
  output_pred[display] = prob