poltextlab commited on
Commit
555f614
·
verified ·
1 Parent(s): bee266b
Files changed (1) hide show
  1. interfaces/cap_minor.py +1 -1
interfaces/cap_minor.py CHANGED
@@ -75,7 +75,7 @@ def predict(text, model_id, tokenizer_id):
75
  output_pred_minor = {f"[{CAP_MIN_NUM_DICT[i]}] {CAP_MIN_LABEL_NAMES[CAP_MIN_NUM_DICT[i]]}": probs[i] for i in np.argsort(probs)[::-1]}
76
  output_pred_major = convert_minor_to_major(np.argsort(probs)[::-1])
77
  output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.</p>'
78
- return output_pred_minor, output_pred_major output_info
79
 
80
  def predict_cap(text, language, domain):
81
  domain = domains[domain]
 
75
  output_pred_minor = {f"[{CAP_MIN_NUM_DICT[i]}] {CAP_MIN_LABEL_NAMES[CAP_MIN_NUM_DICT[i]]}": probs[i] for i in np.argsort(probs)[::-1]}
76
  output_pred_major = convert_minor_to_major(np.argsort(probs)[::-1])
77
  output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.</p>'
78
+ return output_pred_minor, output_pred_major, output_info
79
 
80
  def predict_cap(text, language, domain):
81
  domain = domains[domain]