Update code example in readme to use `device` variable instead of hard coded `cuda` a62f674 verified hl-tburns commited on Jul 23, 2024