top5 normalization for minor codes

interfaces/cap_minor_media.py

@@ -116,7 +116,7 @@ def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
         # Restrict to valid minor codes
         valid_indices = [minor_id_to_index[mid] for mid in valid_minor_ids if mid in minor_id_to_index]
         filtered_probs = {minor_index_to_id[i]: float(minor_probs[0][i]) for i in valid_indices}
-        filtered_probs = normalize_probs(filtered_probs)
+        filtered_probs = normalize_probs(filtered_probs, n=5)
         
         output_pred = {
             f"[{top_major_id}] {CAP_MEDIA_LABEL_NAMES[top_major_id]} [{k}] {CAP_MIN_LABEL_NAMES[k]}": v