debug prints

interfaces/cap_minor_media.py

@@ -95,15 +95,20 @@ def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
     top_major_id = major_index_to_id[top_major_index]
 
     # Default: show major topic predictions
+    print(major_probs_np) # debug
     filtered_probs = {
         i: float(major_probs_np[i])
         for i in np.argsort(major_probs_np)[::-1]
     }
+    print(filtered_probs) # debug
     filtered_probs = normalize_probs(filtered_probs)
+    print(filtered_probs) # debug
+        
     output_pred = {
         f"[{major_index_to_id[k]}] {CAP_MEDIA_LABEL_NAMES[filtered_probs[k]]}": v
         for k, v in sorted(filtered_probs.items(), key=lambda item: item[1], reverse=True)
     }
+    print(output_pred) # debug
 
     # If eligible for minor prediction
     if top_major_id in major_to_minor_map: