Update README.md

README.md

@@ -28,14 +28,12 @@ This model is an attempt to simplify code understanding by generating line by li
 
 ```py
 from transformers import AutoTokenizer, T5ForConditionalGeneration, SummarizationPipeline
-import torch
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 pipeline = SummarizationPipeline(
     model=T5ForConditionalGeneration.from_pretrained("sagard21/python-code-explainer"),
     tokenizer=AutoTokenizer.from_pretrained("sagard21/python-code-explainer", skip_special_tokens=True),
-    device=device
 )
 
 raw_code = """