Spaces:
Running
Running
fix typo
Browse files- 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]
|