pragnakalp commited on
Commit
8bf8bcd
1 Parent(s): 8b9fde5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, 'models/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, 'models/t5_model')
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