cointegrated Den4ikAI commited on
Commit
ba0a6f7
1 Parent(s): c9b878b

Update README.md (#1)

Browse files

- Update README.md (364914481be3d71a110e6ff598bea3c94d2014bf)


Co-authored-by: Denis Petrov <Den4ikAI@users.noreply.huggingface.co>

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)