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