Yes. Punishment is for their action but they should not be humiliated other than what actually the punishment is. They can stop eating non_veg if they are not sure whether its halal or haram.. not being in prison so many muslims do the same..