Update README.md

README.md

@@ -10,6 +10,7 @@ license: MIT
 ## How to use
 
 ```python
+import torch
 from transformers import PreTrainedTokenizerFast
 from transformers import BartForConditionalGeneration
 
@@ -22,7 +23,7 @@ model = BartForConditionalGeneration.from_pretrained('gogamza/kobart-summarizati
 text = "과거를 떠올려보자. 방송을 보던 우리의 모습을..."
 
 raw_input_ids = tokenizer.encode(text)
-input_ids = [tokenizer.bos_token_id] + \
+input_ids = [tokenizer.bos_token_id] + \\
     raw_input_ids + [tokenizer.eos_token_id]
 summary_ids = model.generate(torch.tensor([input_ids]),
                                 max_length=150,