display major topic as well

interfaces/cap_minor_media.py

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