update cluster sort

app.py

@@ -447,6 +447,8 @@ def make_cluster_plot(eigvecs, images, h=64, w=64, progess_start=0.6, advanced=F
         mask = (heatmap > top_p).float()
         # take top 3 masks only
         mask_sort_values = mask.mean((1, 2))
+        _sort_value2 = (heatmap > 0.1).float().mean((1, 2)) * 0.1
+        mask_sort_values += _sort_value2
         mask_sort_idx = torch.argsort(mask_sort_values, descending=True)
         mask = mask[mask_sort_idx[:3]] 
         sort_values.append(mask.mean().item())