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