Den4ikAI commited on
Commit
3649144
1 Parent(s): c9b878b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -17,6 +17,7 @@ This is a model for abstractive Russian summarization, based on [cointegrated/ru
17
  It can be used as follows:
18
 
19
  ```python
 
20
  from transformers import T5ForConditionalGeneration, T5Tokenizer
21
  MODEL_NAME = 'cointegrated/rut5-base-absum'
22
  model = T5ForConditionalGeneration.from_pretrained(MODEL_NAME)
 
17
  It can be used as follows:
18
 
19
  ```python
20
+ import torch
21
  from transformers import T5ForConditionalGeneration, T5Tokenizer
22
  MODEL_NAME = 'cointegrated/rut5-base-absum'
23
  model = T5ForConditionalGeneration.from_pretrained(MODEL_NAME)