Update README.md

README.md

@@ -18,7 +18,7 @@ Here's a quick example of using this model:
 
 ```py
 from transformers import AutoTokenizer, AutoModelWithLMHead
-  
+
 tokenizer = AutoTokenizer.from_pretrained("Sentdex/GPyT")
 model = AutoModelWithLMHead.from_pretrained("Sentdex/GPyT")
 
@@ -27,8 +27,8 @@ inp = """import"""
 
 newlinechar = "<N>"
 converted = inp.replace("\n", newlinechar)
-tokenized = tokenizer.encode(converted, return_tensors='pt').to("cuda")
-resp = model.generate(tokenized).to("cuda")
+tokenized = tokenizer.encode(converted, return_tensors='pt')
+resp = model.generate(tokenized)
 
 decoded = tokenizer.decode(resp[0])
 reformatted = decoded.replace("<N>","\n")