added extra condition to filter out the one with highest prob if the model should return a list fa5eb33 ArpitaShenoy commited on Oct 3, 2022