add device to code

README.md

@@ -380,16 +380,18 @@ Install/update transformers `pip install -U transformers`
 Summarize text with pipeline:
 
 ```python
+import torch
 from transformers import pipeline
 
 summarizer = pipeline(
-    'summarization',
-    'pszemraj/long-t5-tglobal-base-16384-book-summary',
-    )
+    "summarization",
+    "pszemraj/long-t5-tglobal-base-16384-book-summary",
+    device=0 if torch.cuda.is_available() else -1,
+)
 long_text = "Here is a lot of text I don't want to read. Replace me"
 
 result = summarizer(long_text)
-print(result[0]['summary_text'])
+print(result[0]["summary_text"])
 ```
 
 Pass [other parameters related to beam search textgen](https://huggingface.co/blog/how-to-generate) when calling `summarizer` to get even higher quality results.