Update README.md

README.md

@@ -23,7 +23,7 @@ Use the code below to get started with CodonTransformer:
 from transformers import AutoTokenizer, BigBirdForMaskedLM
 
 tokenizer = AutoTokenizer.from_pretrained("adibvafa/CodonTransformer")
-model = BigBirdForMaskedLM.from_pretrained("adibvafa/CodonTransformer").to("cuda" if torch.cuda.is_available() else "cpu")
+model = BigBirdForMaskedLM.from_pretrained("adibvafa/CodonTransformer")
 ```
 
 Sample demo is shown at https://github.com/Adibvafa/CodonTransformer/blob/main/CodonTransformerDemo.ipynb