mask = prob_original >= 0.35