Update README.md

README.md

@@ -34,6 +34,7 @@ This is the model for abstractive summarization for Russian based on [rugpt3medi
 Colab: [link](https://colab.research.google.com/drive/1re5E26ZIDUpAx1gOCZkbF3hcwjozmgG0)
 
 ```python
+import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 model_name = "IlyaGusev/rugpt3medium_sum_gazeta"
@@ -50,7 +51,7 @@ text_tokens = tokenizer(
     truncation=True
 )["input_ids"]
 input_ids = text_tokens + [tokenizer.sep_token_id]
-input_ids = torch.LongTensor([input_ids]).to(device)
+input_ids = torch.LongTensor([input_ids])
 
 output_ids = model.generate(
     input_ids=input_ids,