typo

interfaces/cap_minor_media.py

@@ -105,7 +105,7 @@ def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
     print(filtered_probs) # debug
         
     output_pred = {
-        f"[{major_index_to_id[k]}] {CAP_MEDIA_LABEL_NAMES[filtered_probs[k]]}": v
+        f"[{major_index_to_id[k]}] {CAP_MEDIA_LABEL_NAMES[k]}": v
         for k, v in sorted(filtered_probs.items(), key=lambda item: item[1], reverse=True)
     }
     print(output_pred) # debug