* Switch model and fix demo * fix remaining demos --------- Co-authored-by: Abubakar Abid <abubakar@huggingface.co>