Spaces:
Running
Running
update ordering
Browse files- interfaces/cap_minor.py +1 -1
interfaces/cap_minor.py
CHANGED
@@ -79,7 +79,7 @@ def predict(text, model_id, tokenizer_id):
|
|
79 |
output_pred = {f"[{'999' if str(CAP_MIN_NUM_DICT[i]) == '999' else str(CAP_MIN_NUM_DICT[i])[:-2]}]{convert_minor_to_major(CAP_MIN_NUM_DICT[i])} [{CAP_MIN_NUM_DICT[i]}]{CAP_MIN_LABEL_NAMES[CAP_MIN_NUM_DICT[i]]}": probs[i] for i in np.argsort(probs)[::-1]}
|
80 |
|
81 |
|
82 |
-
output_pred = dict(sorted(output_pred.items(), key=lambda item: item[1]))
|
83 |
first_n_items = take(5, output_pred.items())
|
84 |
|
85 |
html = ""
|
|
|
79 |
output_pred = {f"[{'999' if str(CAP_MIN_NUM_DICT[i]) == '999' else str(CAP_MIN_NUM_DICT[i])[:-2]}]{convert_minor_to_major(CAP_MIN_NUM_DICT[i])} [{CAP_MIN_NUM_DICT[i]}]{CAP_MIN_LABEL_NAMES[CAP_MIN_NUM_DICT[i]]}": probs[i] for i in np.argsort(probs)[::-1]}
|
80 |
|
81 |
|
82 |
+
output_pred = dict(sorted(output_pred.items(), key=lambda item: item[1], reverse=True))
|
83 |
first_n_items = take(5, output_pred.items())
|
84 |
|
85 |
html = ""
|