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