Update example with torch device

README.md

@@ -337,26 +337,17 @@ model-index:
 - create the pipeline object:
 
 ```python
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+
+import torch
 from transformers import pipeline
 
 hf_name = 'pszemraj/led-base-book-summary'
 
-_model = AutoModelForSeq2SeqLM.from_pretrained(
-                hf_name,
-                low_cpu_mem_usage=True,
-            )
-
-_tokenizer = AutoTokenizer.from_pretrained(
-                hf_name
-            )
-                                           
-
 summarizer = pipeline(
-                    "summarization", 
-                    model=_model, 
-                    tokenizer=_tokenizer
-                )
+    "summarization",
+    hf_name,
+    device=0 if torch.cuda.is_available() else -1,
+)
 ```
 
 - put words into the pipeline object: