Anonumous commited on
Commit
a31378a
1 Parent(s): b3b88bc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -15,8 +15,8 @@ The model is trained on a mix of open summarisation data for the Russian languag
15
  ```python
16
  import torch
17
  from transformers import GPT2Tokenizer, T5ForConditionalGeneration
18
- tokenizer = GPT2Tokenizer.from_pretrained('Anonumous/FRED-T5-SUM',eos_token='</s>')
19
- model = T5ForConditionalGeneration.from_pretrained('Anonumous/FRED-T5-SUM')
20
  device='cuda'
21
  model.to(device)
22
 
 
15
  ```python
16
  import torch
17
  from transformers import GPT2Tokenizer, T5ForConditionalGeneration
18
+ tokenizer = GPT2Tokenizer.from_pretrained('RussianNLP/FRED-T5-Summarizer',eos_token='</s>')
19
+ model = T5ForConditionalGeneration.from_pretrained('RussianNLP/FRED-T5-Summarizer')
20
  device='cuda'
21
  model.to(device)
22