Update README.md

README.md

@@ -27,10 +27,9 @@ load the model into a `pipeline` object:
 ```
 from transformers import pipeline
 import torch
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
 my_chatbot = pipeline('text-generation', 
                       'pszemraj/gpt-peter-2.7B',
-                      device=0 if device == 'cuda' else -1,
+                      device=0 if torch.cuda.is_available() else -1,
                     )
 ```