IlyaGusev
/

rubert_telegram_headlines

encoder-decoder

text2text-generation

Model card Files Files and versions

IlyaGusev commited on Mar 15, 2021

Commit

8f3eeb4

·

1 Parent(s): d04dd66

Update README.md

Files changed (1) hide show

README.md +2 -5

README.md CHANGED Viewed

@@ -56,14 +56,11 @@ print(headline)
 ## Training procedure
 ```python
-import json
-import os
 import random
-import shutil
 import torch
 from torch.utils.data import Dataset
-from tqdm import tqdm
 from transformers import BertTokenizer, EncoderDecoderModel, Trainer, TrainingArguments, logging

 ## Training procedure
 ```python
 import random
 import torch
 from torch.utils.data import Dataset
+from tqdm.notebook import tqdm
 from transformers import BertTokenizer, EncoderDecoderModel, Trainer, TrainingArguments, logging