Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ from pytorch_lightning.loggers import TensorBoardLogger
|
|
10 |
from datasets.dataset_dict import DatasetDict
|
11 |
from transformers import AdamW, T5ForConditionalGeneration, T5TokenizerFast
|
12 |
from tqdm.auto import tqdm
|
13 |
-
from
|
14 |
from transformers import AutoTokenizer
|
15 |
from sentence_transformers import SentenceTransformer
|
16 |
import warnings
|
|
|
10 |
from datasets.dataset_dict import DatasetDict
|
11 |
from transformers import AdamW, T5ForConditionalGeneration, T5TokenizerFast
|
12 |
from tqdm.auto import tqdm
|
13 |
+
from summarizer import SummarizerModel
|
14 |
from transformers import AutoTokenizer
|
15 |
from sentence_transformers import SentenceTransformer
|
16 |
import warnings
|