Spaces:
Runtime error
Runtime error
pragnakalp
commited on
Commit
·
8bf8bcd
1
Parent(s):
8b9fde5
Update app.py
Browse files
app.py
CHANGED
@@ -13,11 +13,11 @@ read = Readability()
|
|
13 |
cwd = os.getcwd()
|
14 |
readablility_nlp.add_pipe(read, last=True)
|
15 |
|
16 |
-
bart_ext_model_path = os.path.join(cwd, '
|
17 |
bart_extractive_model = BartForConditionalGeneration.from_pretrained(bart_ext_model_path)
|
18 |
bart_extractive_tokenizer = BartTokenizer.from_pretrained('facebook/bart-large-cnn')
|
19 |
|
20 |
-
t5_model_path = os.path.join(cwd, '
|
21 |
t5_model = AutoModelWithLMHead.from_pretrained(t5_model_path)
|
22 |
t5_tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
|
23 |
|
|
|
13 |
cwd = os.getcwd()
|
14 |
readablility_nlp.add_pipe(read, last=True)
|
15 |
|
16 |
+
bart_ext_model_path = os.path.join(cwd, 'bart_extractive_model')
|
17 |
bart_extractive_model = BartForConditionalGeneration.from_pretrained(bart_ext_model_path)
|
18 |
bart_extractive_tokenizer = BartTokenizer.from_pretrained('facebook/bart-large-cnn')
|
19 |
|
20 |
+
t5_model_path = os.path.join(cwd, 't5_model')
|
21 |
t5_model = AutoModelWithLMHead.from_pretrained(t5_model_path)
|
22 |
t5_tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
|
23 |
|