oops

interfaces/cap_minor_media.py

@@ -43,7 +43,7 @@ def normalize_probs(probs: dict, n: int):
     probs_ = list(probs.values())
     if len(probs_) > n:
         probs_.sort(reverse=True)
-        probs_ = probs[:5]
+        probs_ = probs_[:5]
     values = np.array(probs_)
     exp_values = np.exp(values)
     sum_exp = np.sum(exp_values)